diff --git a/bin/video2x.py b/bin/video2x.py index 96d1e1e..a6d442a 100644 --- a/bin/video2x.py +++ b/bin/video2x.py @@ -76,16 +76,16 @@ def process_arguments(): upscaler_options = parser.add_argument_group('Upscaler Options') upscaler_options.add_argument('-m', '--method', help='Upscaling method', action='store', default='gpu', choices=['cpu', 'gpu', 'cudnn'], required=True) upscaler_options.add_argument('-d', '--driver', help='Waifu2x driver', action='store', default='waifu2x_caffe', choices=['waifu2x_caffe', 'waifu2x_converter']) - upscaler_options.add_argument('-y', '--model_dir', help='Folder containing model JSON files', action='store', default=None) + upscaler_options.add_argument('-y', '--model_dir', help='Folder containing model JSON files', action='store') upscaler_options.add_argument('-t', '--threads', help='Number of threads to use for upscaling', action='store', type=int, default=5) upscaler_options.add_argument('-c', '--config', help='Video2X config file location', action='store', default='{}\\video2x.json'.format(os.path.dirname(os.path.abspath(__file__)))) - upscaler_options.add_argument('-b', '--batch', help='Enable batch mode (select all default values to questions)', action='store_true', default=False) + upscaler_options.add_argument('-b', '--batch', help='Enable batch mode (select all default values to questions)', action='store_true') # scaling options scaling_options = parser.add_argument_group('Scaling Options') - scaling_options.add_argument('--width', help='Output video width', action='store', type=int, default=False) - scaling_options.add_argument('--height', help='Output video height', action='store', type=int, default=False) - scaling_options.add_argument('-r', '--ratio', help='Scaling ratio', action='store', type=float, default=False) + scaling_options.add_argument('--width', help='Output video width', action='store', type=int) + scaling_options.add_argument('--height', help='Output video height', action='store', type=int) + scaling_options.add_argument('-r', '--ratio', help='Scaling ratio', action='store', type=float) # parse arguments return parser.parse_args()