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()