test_harris_numpi.py

Go to the documentation of this file.

# *****************************************************************************
# *                            cloudFPGA
# *                Copyright 2016 -- 2022 IBM Corporation
# * Licensed under the Apache License, Version 2.0 (the "License");
# * you may not use this file except in compliance with the License.
# * You may obtain a copy of the License at
# *
# *     http://www.apache.org/licenses/LICENSE-2.0
# *
# * Unless required by applicable law or agreed to in writing, software
# * distributed under the License is distributed on an "AS IS" BASIS,
# * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# * See the License for the specific language governing permissions and
# * limitations under the License.
# *----------------------------------------------------------------------------
 
 
 
import sys
import os
import numpy as np
import cv2
 
trieres_lib=os.environ['cFpRootDir'] + "HOST/vision/harris/languages/python/build"
sys.path.append(trieres_lib)
 
import _trieres_harris_numpi
 
# size of image to be processed on fpga (the bitstream should be already fixed to this)
height = width = 256
 
num_frame = 1
 
# path  
image_in_filename = os.environ['cFpRootDir'] + "ROLE/vision/hls/harris/test/512x512.png"
image_out_filename = image_in_filename + "_fpga_points_out_frame_" + str(num_frame) + ".png"
 
for i in range(10):
    # Reading an image in unchanged mode 
    image = cv2.imread(image_in_filename, cv2.IMREAD_UNCHANGED) 
 
    # Converting to grayscale
    image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
 
    # Adjusting the image file if needed
    if ((image.shape[0] != height) or (image.shape[1] != width)):
        print("WARNING: The image was resized from [", image.shape[0] , " x ", image.shape[1] , "] to [", height  , " x ", width, "]")    
        dim = (width, height) 
        image = cv2.resize(image, dim, interpolation = cv2.INTER_LINEAR) 
 
    # Flattening the image from 2D to 1D
    image = image.flatten()
 
    total_size = height * width
 
    input_array = image
 
#for i in range(100):
 
    output_array = _trieres_harris_numpi.harris(input_array, total_size, "10.12.200.234", "2719")
 
    # Convert 1D array to a 2D numpy array of 2 rows and 3 columns
    output_array_2d = np.reshape(output_array, (height, width))
 
cv2.imwrite(image_out_filename, output_array_2d)
print("INFO: the output file is saved at : " + image_out_filename)