Browse Source

fix video reader (#2967)

Sunflower7788 10 tháng trước cách đây
mục cha
commit
9342df2d84
1 tập tin đã thay đổi với 3 bổ sung3 xóa
  1. 3 3
      paddlex/inference/utils/io/readers.py

+ 3 - 3
paddlex/inference/utils/io/readers.py

@@ -318,9 +318,9 @@ class OpenCVVideoReaderBackend(_VideoReaderBackend):
         self._cap_release()
 
     def _cap_open(self, video_path):
-        self.cap_init_args.pop("num_seg")
-        self.cap_init_args.pop("seg_len")
-        self.cap_init_args.pop("sample_type")
+        self.cap_init_args.pop("num_seg", None)
+        self.cap_init_args.pop("seg_len", None)
+        self.cap_init_args.pop("sample_type", None)
         self._cap = cv2.VideoCapture(video_path, **self.cap_init_args)
         if not self._cap.isOpened():
             raise RuntimeError(f"Failed to open {video_path}")