Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2026-02-24 04:19:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add tqdm

Browse Source
This commit is contained in:
hzwer
2020-11-15 17:01:29 +08:00
parent 514fd751ef
commit 63947bc097
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
inference_mp4_2x.py
View File

@@ -3,6 +3,7 @@ import cv2
import torch
import argparse
import numpy as np
from tqdm import tqdm
from torch.nn import functional as F
from model.RIFE import Model
@@ -26,7 +27,9 @@ ph = ((h - 1) // 32 + 1) * 32
pw = ((w - 1) // 32 + 1) * 32
padding = (0, pw - w, 0, ph - h)
fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
print('{}.mp4'.format(args.video[:-4]))
tot_frame = videoCapture.get(cv2.CAP_PROP_FRAME_COUNT)
pbar = tqdm(total=tot_frame)
print('{}.mp4, {} frames in total, {}FPS to {}FPS'.format(args.video[:-4], tot_frame, fps, fps * 2))
if args.montage:
output = cv2.VideoWriter('{}_2x.mp4'.format(args.video[:-4]), fourcc, fps * 2, (2*w, h))
else:
@@ -52,8 +55,10 @@ while success:
else:
output.write(lastframe)
output.write(mid[:h, :w])
pbar.update(1)
if args.montage:
output.write(np.concatenate((lastframe, lastframe), 1))
else:
output.write(lastframe)
pbar.close()
output.release()