diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index 6ea92ed3d8e63f7403b97e1da739741d38166760..99d5424c084cf56b12e93fbc14bcfad55180d499 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -534,6 +534,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
@@ -550,7 +552,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,