def kinfu_demo(): parser = ArgumentParser() parser.add_argument( "-i", "--input", help="Required. Path to folder with a input image file", required=True, type=str) parser.add_argument("-t", "--large_kinfu", help="Required. Name of KinFu type", required=False, type=str) parser.add_argument("-ocl", "--use_opencl", help="Required. Flag of OpenCL use", required=False, type=int, default=1) args = parser.parse_args() print("Args: ", args) cv.ocl.setUseOpenCL(args.use_opencl) if (args.large_kinfu == None or args.large_kinfu == "0"): params = cv.kinfu_Params.defaultParams() kf = cv.kinfu_KinFu.create(params) elif (args.large_kinfu == "1"): params = cv.kinfu_Params.hashTSDFParams(False) kf = cv.kinfu_KinFu.create(params) else: raise ValueError("Incorrect kinfu type name") depth_list = get_depth_list(args.input) for path in depth_list: image = cv.imread(path, cv.IMREAD_ANYDEPTH) (height, width) = image.shape cv.imshow('input', image) size = height, width, 4 cvt8 = np.zeros(size, dtype=np.uint8) if not kf.update(image): kf.reset() else: kf.render(cvt8) cv.imshow('render', cvt8) cv.pollKey() cv.waitKey(0)
def show_img(img, win_title='-', wait4key=True): cv2.setWindowTitle(winname=WIN_NAME, title=win_title) cv2.imshow(WIN_NAME, img) if wait4key: key = chr(cv2.waitKey()) else: key = cv2.pollKey() if key >= 0: key = chr(key) return key
img_rep = represent.represent(img_aln, input_shape) retrieved_img = img_rep[0][:, :, ::-1] plt.imshow(retrieved_img) plt.show() return "Registered" video_capture = cv2.VideoCapture(0) while True: ret, frame = video_capture.read() detections = detect.getDetections(frame, face_detector) if detections != None: orig_frame = frame.copy() for detection in detections: x, y, w, h = detection["box"] cv2.rectangle(frame, (x, y + h), (x + w, y), (0, 0, 255), 2) cv2.imshow('Video', frame) keyPress = cv2.pollKey() & 0xFF if keyPress == ord('r'): register(face_detector, model, input_shape, orig_frame, detections) elif keyPress == ord('q'): # Hit 'q' on the keyboard to quit! print("..... Quiting") break video_capture.release() cv2.destroyAllWindows()
while 1: t1 = time.perf_counter() kImage += 1 (img_buffer, img_info) = ebus.getImage(timeoutMS) # print(img_info) t2 = time.perf_counter() info = expand_img_info_tuple(img_info) # print(info) # TODO: processing goes here # hint: map the image into a numpy array by doing np.frombuffer(img_buffer, np.uint16) img_np = np.frombuffer(img_buffer, np.uint16) img_np = img_np.reshape(info['Height'], info['Width']) t3 = time.perf_counter() cv2.imshow('Press any key to stop', img_np) if cv2.pollKey() != -1: break t4 = time.perf_counter() print( "kImage=%d, t(getImage)=%.3f ms, t(numpy)=%.3f ms, t(cv)=%.3f ms, min, max: %d, %d" % (kImage, (t2 - t1) * 1e3, (t3 - t2) * 1e3, (t4 - t3) * 1e3, np.min(img_np), np.max(img_np))) ebus.releaseImage() ##################### End image processing loop ##################### # ebus.openDeviceSerialPort() ebus.writeSerialPort('test string\n') result = ebus.readSerialPort(10, 500) print(result) ebus.closeDeviceSerialPort()