image = cv2.resize(image, tuple(image_shape[:2][::-1])) image = np.expand_dims(image, axis=0) # K.set_session(session) extractor = MarsExtractorAPI('mars_api', True) ip = extractor.get_in_pipe() # op = extractor.get_out_pipe() extractor.use_session_runner(session_runner) session_runner.start() extractor.run() ret_pipe = Pipe() # for i in range(1000): i = 0 while True: ret, image = cap.read() if not ret: continue ip.push(Inference(image, ret_pipe, {})) ret, feature_inference = ret_pipe.pull() if ret: print(feature_inference.get_result().shape) else: ret_pipe.wait() session_runner.stop()