github
/
arXiv2020-RIFE
mirror of https://github.com/hzwer/arXiv2020-RIFE
1
Fork 0
Code Releases Activity

Update inference_video.py

This commit is contained in:
hzwer 2022-12-05 11:29:22 +08:00 committed by GitHub
parent 051fa1c8c0
commit 2477b532c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference_video.py
View File

@ -167,7 +167,7 @@ def build_read_buffer(user_args, read_buffer, videogen):
try:
for frame in videogen:
if not user_args.img is None:
frame = cv2.imread(os.path.join(user_args.img, frame))[:, :, ::-1].copy()
frame = cv2.imread(os.path.join(user_args.img, frame), cv2.IMREAD_UNCHANGED)[:, :, ::-1].copy()
if user_args.montage:
frame = frame[:, left: left + w]
read_buffer.put(frame)