diff --git a/inference_mp4_2x.py b/inference_mp4_2x.py index 8b50115..edcb557 100644 --- a/inference_mp4_2x.py +++ b/inference_mp4_2x.py @@ -44,7 +44,7 @@ padding = (0, pw - w, 0, ph - h) tot_frame = videoCapture.get(cv2.CAP_PROP_FRAME_COUNT) print('{}.mp4, {} frames in total, {}FPS to {}FPS'.format(args.video[:-4], tot_frame, fps, args.fps)) pbar = tqdm(total=tot_frame) -cnt = 0 +cnt = 1 if args.montage: frame = frame[:, left: left + w] while success: diff --git a/inference_mp4_4x.py b/inference_mp4_4x.py index 7dfbd09..29e54e3 100644 --- a/inference_mp4_4x.py +++ b/inference_mp4_4x.py @@ -44,7 +44,7 @@ padding = (0, pw - w, 0, ph - h) tot_frame = videoCapture.get(cv2.CAP_PROP_FRAME_COUNT) print('{}.mp4, {} frames in total, {}FPS to {}FPS'.format(args.video[:-4], tot_frame, fps, args.fps)) pbar = tqdm(total=tot_frame) -cnt = 0 +cnt = 1 if args.montage: frame = frame[:, left: left + w] while success: