We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e233e92 commit ee290feCopy full SHA for ee290fe
run.py
@@ -227,6 +227,9 @@ def main():
227
# print("normal frame:", idx_frame)
228
LR_img = images[idx+i_frame] if is_video else cv2.imread(images[idx_frame+i_frame], cv2.IMREAD_COLOR)
229
230
+ if not only_y:
231
+ LR_img = cv2.cvtColor(LR_img, cv2.COLOR_BGR2RGB)
232
+
233
# get the bicubic upscale of the center frame to concatenate for SR
234
if only_y and i_frame == idx_center:
235
if args.denoise:
@@ -311,7 +314,7 @@ def main():
311
314
else:
312
315
sr_img = SR
313
316
- sr_img = util.tensor2np(sr_img, rgb2bgr=only_y) # uint8
317
+ sr_img = util.tensor2np(sr_img) # uint8
318
319
if not is_video:
320
# save images
0 commit comments