From ab2f982a84445598ecbec60f663c34e55631f172 Mon Sep 17 00:00:00 2001 From: k4yt3x Date: Mon, 11 May 2020 04:33:38 -0400 Subject: [PATCH] fixed waifu2x-caffe parsing error --- src/upscaler.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/upscaler.py b/src/upscaler.py index 22bd4c1..21ce7a4 100755 --- a/src/upscaler.py +++ b/src/upscaler.py @@ -222,9 +222,9 @@ class Upscaler: raise e # waifu2x-caffe scale_ratio, scale_width and scale_height check - if self.driver == 'waifu2x-caffe': - if (driver_settings['scale_width'] != 0 and driver_settings['scale_width'] == 0 or - driver_settings['scale_width'] == 0 and driver_settings['scale_width'] != 0): + if self.driver == 'waifu2x_caffe': + if (driver_settings['scale_width'] != 0 and driver_settings['scale_height'] == 0 or + driver_settings['scale_width'] == 0 and driver_settings['scale_height'] != 0): Avalon.error('Only one of scale_width and scale_height is specified for waifu2x-caffe') raise AttributeError('only one of scale_width and scale_height is specified for waifu2x-caffe')