From e411c8cf4eb7b9acd892b9b090db3bcac38401da Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Sun, 15 Nov 2020 18:04:47 +0800
Subject: [PATCH] Try to fix 60FPS

---
 inference_mp4_2x.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/inference_mp4_2x.py b/inference_mp4_2x.py
index c8153fb..25321db 100644
--- a/inference_mp4_2x.py
+++ b/inference_mp4_2x.py
@@ -16,7 +16,6 @@ if torch.cuda.is_available():
 parser = argparse.ArgumentParser(description='Interpolation for a pair of images')
 parser.add_argument('--video', dest='video', required=True)
 parser.add_argument('--montage', dest='montage', action='store_true', help='montage origin video')
-parser.add_argument('--fps', dest='fps', type=int, default=60)
 args = parser.parse_args()
 
 model = Model()
@@ -33,7 +32,7 @@ pw = ((w - 1) // 32 + 1) * 32
 padding = (0, pw - w, 0, ph - h)
 fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
 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))
+print('{}.mp4, {} frames in total, {}FPS'.format(args.video[:-4], tot_frame, fps))
 pbar = tqdm(total=tot_frame)
 if args.montage:
     output = cv2.VideoWriter('{}_2x.mp4'.format(args.video[:-4]), fourcc, args.fps, (2*w, h))