Ver Fonte

fix video reader (#2967)

Sunflower7788 há 9 meses atrás
pai
commit
9342df2d84
1 ficheiros alterados com 3 adições e 3 exclusões
  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}")