| | import argparse |
| |
|
| | parser = argparse.ArgumentParser(description='PAN') |
| |
|
| | |
| | parser.add_argument('--n_threads', type=int, default=6, |
| | help='Number of threads for data loading') |
| | parser.add_argument('--ipu', action='store_true', help='Use Ipu for interence.') |
| |
|
| | |
| | parser.add_argument('--dir_data', type=str, default='dataset/', |
| | help='Dataset directory') |
| | parser.add_argument('--data_test', type=str, default='Set5', |
| | help='Test dataset name') |
| | parser.add_argument('--ext', type=str, default='sep', |
| | help='Dataset file extension') |
| | parser.add_argument('--scale', type=str, default='2', |
| | help='Super resolution scale') |
| | parser.add_argument('--rgb_range', type=int, default=255, |
| | help='Maximum value of RGB') |
| | parser.add_argument('--n_colors', type=int, default=3, |
| | help='Number of color channels to use') |
| | parser.add_argument('--onnx_path', type=str, default='PAN_int8.onnx', |
| | help='Path to onnx model') |
| | parser.add_argument('--provider_config', |
| | type=str, |
| | default="vaip_config.json", |
| | help="Path of the config file for seting provider_options.") |
| |
|
| |
|
| | args = parser.parse_args() |
| |
|
| | args.scale = list(map(lambda x: int(x), args.scale.split('+'))) |
| | args.data_test = args.data_test.split('+') |
| |
|
| |
|
| | for arg in vars(args): |
| | if vars(args)[arg] == 'True': |
| | vars(args)[arg] = True |
| | elif vars(args)[arg] == 'False': |
| | vars(args)[arg] = False |
| |
|
| |
|
