Classes
class	FPS

class	DummyTask

Functions
def	main ()

Variables
string	video_common_lib = os.environ['cFpRootDir'] + "HOST/vision/common/languages/python/var"

string	trieres_lib = os.environ['cFpRootDir'] + "HOST/vision/harris/languages/python/build"

int	height = 512

int	total_size = height * width

Function Documentation

◆ main()

def test_harris_numpi_video_threaded.main ( )

Definition at line 105 of file test_harris_numpi_video_threaded.py.

 def main():
     import sys
  
     try:
         fn = sys.argv[1]
     except:
         fn = 0
     cap = video.create_capture(fn)
     fps = FPS().start()
  
     video_name = str(fn)+"_out.avi"
     video_out = cv.VideoWriter(video_name, cv.VideoWriter_fourcc('M','J','P','G'), 10, (width,height))
     
     fpgas = deque([ ["10.12.200.3"   , "2718"],
                     ["10.12.200.165" , "2719"]])
  
     
     def process_frame(frame, t0, accel_mode, fpga):
         if accel_mode:
             print("Will execute on fpga with ip:port: "+fpga[0]+":"+fpga[1])
             # some intensive computation...
             # frame = cv.medianBlur(frame, 19)
             # Converting to grayscale
             frame = cv.cvtColor(frame, cv.COLOR_RGB2GRAY)
             # Adjusting the image file if needed
             if ((frame.shape[0] != height) or (frame.shape[1] != width)):
                 print("WARNING: The image was resized from [", frame.shape[0] , " x ", frame.shape[1] , "] to [", height  , " x ", width, "]")
             dim = (width, height)
             frame = cv.resize(frame, dim, interpolation = cv.INTER_LINEAR)
             # Flattening the image from 2D to 1D
             image = frame.flatten()
             output_array = _trieres_harris_numpi.harris(image, total_size, fpga[0], fpga[1])
             # Convert 1D array to a 2D numpy array 
             frame = np.reshape(output_array, (height, width))
             print("Declare free the fpga: "+str(fpga))
             fpgas.appendleft(fpga)
         else:
             frame = cv.medianBlur(frame, 19)
             frame = cv.medianBlur(frame, 19)
             frame = cv.medianBlur(frame, 19)
         return frame, t0
  
     threadn = 4 #cv.getNumberOfCPUs()
     pool = ThreadPool(processes = threadn)
     pending = deque()
  
     threaded_mode = True
     accel_mode = True
     latency = StatValue()
     frame_interval = StatValue()
     last_frame_time = clock()
     while True:
         while len(pending) > 0 and pending[0].ready() and len(fpgas) > 0:
             res, t0 = pending.popleft().get()
             latency.update(clock() - t0)
 #            video_out.write(res)
             draw_str(res, (20, 20), "threaded       :  " + str(threaded_mode))
             draw_str(res, (20, 40), "cloudFPA       :  " + str(accel_mode))
             draw_str(res, (20, 60), "latency        :  %.1f ms" % (latency.value*1000))
             draw_str(res, (20, 80), "frame interval :  %.1f ms" % (frame_interval.value*1000))
             draw_str(res, (20, 100), "FPS           :  %.1f" % (1.0/frame_interval.value))
             #cv.imshow('threaded video', res)
         if len(pending) < threadn and len(fpgas) != 0:
             _ret, frame = cap.read()
             if _ret is False:
                 print("Reached EOF.")
                 print("Saved video: " + video_name)
                 video_out.release()
                 break
             else:
                 video_out.write(frame)
             t = clock()
             frame_interval.update(t - last_frame_time)
             last_frame_time = t
             # update the FPS counter
             fps.update()            
             if accel_mode:
                 fpga = fpgas.popleft()
             else:
                 fpga = 0
             if threaded_mode:
                 task = pool.apply_async(process_frame, (frame.copy(), t, accel_mode, fpga))
             else:
                 task = DummyTask(process_frame(frame, t, accel_mode, fpga))
             pending.append(task)
         else:
             if accel_mode:
                 print("Waiting for a free fpga")
             
         ch = cv.waitKey(1)
         if ch == ord(' '):
             threaded_mode = not threaded_mode
         if ch == ord('f'):
             accel_mode = not accel_mode
         if ch == 27:
             break
  
  
     print('Done')
  
     # stop the timer and display FPS information
     fps.stop()
     print("[INFO] elasped time: {:.2f}".format(fps.elapsed()))
     print("[INFO] approx. FPS: {:.2f}".format(fps.fps()))
  