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

Fix inference

Browse Source
This commit is contained in:
hzwer
2020-11-16 17:55:21 +08:00
parent 5f54e36a85
commit 421588e925
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
inference_img.py
View File

@@ -42,4 +42,4 @@ for i in range(args.times):
if not os.path.exists('output'):
os.mkdir('output')
for i in range(len(img_list)):
cv2.imwrite('output/img{}.png'.format(i), img_list[i][0].numpy().transpose(1, 2, 0)[:h, :w] * 255)
cv2.imwrite('output/img{}.png'.format(i), img_list[i][0].cpu().numpy().transpose(1, 2, 0)[:h, :w] * 255)