From a83249c670cc3f1a195d072e6f5533bd33d3c499 Mon Sep 17 00:00:00 2001 From: k4yt3x Date: Mon, 11 May 2020 04:38:33 -0400 Subject: [PATCH] fixed waifu2x-caffe pixel format check typo --- src/upscaler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/upscaler.py b/src/upscaler.py index 21ce7a4..c342e04 100755 --- a/src/upscaler.py +++ b/src/upscaler.py @@ -479,7 +479,7 @@ class Upscaler: framerate = float(Fraction(video_info['streams'][video_stream_index]['r_frame_rate'])) fm.pixel_format = video_info['streams'][video_stream_index]['pix_fmt'] - if self.driver == 'waifu2x-caffe': + if self.driver == 'waifu2x_caffe': # get a dict of all pixel formats and corresponding bit depth pixel_formats = fm.get_pixel_formats()