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

Count frame

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

4
inference_video_parallel.py
View File

@@ -108,9 +108,11 @@ print('{}.{}, {} frames in total, {}FPS to {}FPS'.format(video_path_wo_ext, args
pbar = tqdm(total=tot_frame)
img_list = []
_thread.start_new_thread(clear_buffer, (args, ))
tot_frame -= 1
for frame in videogen:
tot_frame -= 1
img_list.append(frame)
if len(img_list) == 5 or (not success and len(img_list) > 1):
if len(img_list) == 5 or tot_frame == 0:
imgs = torch.from_numpy(np.transpose(img_list, (0, 3, 1, 2))).to(device, non_blocking=True).float() / 255.
I0 = imgs[:-1]
I1 = imgs[1:]