Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2025-12-29 00:21:07 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix first frame

Browse Source
This commit is contained in:
hzwer
2020-11-23 11:29:07 +08:00
parent e9db322d40
commit 3cf2fbe78d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
inference_video_parallel.py
View File

@@ -39,7 +39,7 @@ 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()
frame = next(videogen)
h, w, _ = frame.shape
if args.fps is None:
args.fps = fps * args.times
@@ -55,7 +55,7 @@ else:
cnt = 0
skip_frame = 1
buffer = Queue()
buffer.put(frame)
def write_frame(i0, infs, i1, p, user_args):
global skip_frame, cnt
for i in range(i0.shape[0]):