Check video validity early on.

Preliminary check to protect users against #557.
DeepLabCut · Jan 21, 2020 · 2e6a70d · 2e6a70d
1 parent e6eecc3
commit 2e6a70d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deeplabcut/pose_estimation_tensorflow/predict_videos.py b/deeplabcut/pose_estimation_tensorflow/predict_videos.py
@@ -470,7 +470,8 @@ def AnalyzeVideo(video,DLCscorer,DLCscorerlegacy,trainFraction,cfg,dlc_cfg,sess,
     if notanalyzed:
         print("Loading ", video)
         cap=cv2.VideoCapture(video)
-
+        if not cap.isOpened():
+            raise IOError('Video could not be opened. Please check that the path is valid.')
         fps = cap.get(5) #https://docs.opencv.org/2.4/modules/highgui/doc/reading_and_writing_images_and_video.html#videocapture-get
         nframes = int(cap.get(7))
         duration=nframes*1./fps