From 5f59a4b506d86a1a595ceb25636673fea8a20c9f Mon Sep 17 00:00:00 2001
From: cnn <liuhui29@baidu.com>
Date: Tue, 25 May 2021 21:07:26 +0800
Subject: [PATCH] fix bug of predict video (#3158)

---
 deploy/python/infer.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index fd1d16d59..9552e1bba 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -531,6 +531,8 @@ def predict_video(detector, camera_id):
         capture = cv2.VideoCapture(FLAGS.video_file)
         video_name = os.path.split(FLAGS.video_file)[-1]
     fps = 30
+    frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
+    print('frame_count', frame_count)
     width = int(capture.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(capture.get(cv2.CAP_PROP_FRAME_HEIGHT))
     # yapf: disable
@@ -547,7 +549,7 @@ def predict_video(detector, camera_id):
             break
         print('detect frame:%d' % (index))
         index += 1
-        results = detector.predict(frame, FLAGS.threshold)
+        results = detector.predict([frame], FLAGS.threshold)
         im = visualize_box_mask(
             frame,
             results,
-- 
GitLab