diff --git a/inference_video.py b/inference_video.py index 132b3ef..9dc759f 100644 --- a/inference_video.py +++ b/inference_video.py @@ -38,6 +38,7 @@ model.device() videoCapture = cv2.VideoCapture(args.video) fps = np.round(videoCapture.get(cv2.CAP_PROP_FPS)) tot_frame = videoCapture.get(cv2.CAP_PROP_FRAME_COUNT) +videoCapture.release() if args.fps is None: args.fps = fps * args.exp videogen = skvideo.io.vreader(args.video) diff --git a/inference_video_parallel.py b/inference_video_parallel.py index f5ce262..7eb9891 100644 --- a/inference_video_parallel.py +++ b/inference_video_parallel.py @@ -37,6 +37,7 @@ model.device() videoCapture = cv2.VideoCapture(args.video) fps = np.round(videoCapture.get(cv2.CAP_PROP_FPS)) +videoCapture.release() videogen = skvideo.io.vreader(args.video) success, frame = videoCapture.read() h, w, _ = frame.shape