diff --git a/src/image_demo.py b/src/image_demo.py index 6c993e6..bd12beb 100644 --- a/src/image_demo.py +++ b/src/image_demo.py @@ -14,7 +14,7 @@ class ImageDemoOptions(): self.parser.add_argument('--model_paths', '-n', nargs='+', type=str, default=["../models/last_transfered_net.pth","../models/last_transfered_lut.pth"], help="Model paths for comparison") self.parser.add_argument('--hr_image_path', '-a', type=str, default="../data/Set14/HR/monarch.png", help="HR image path") self.parser.add_argument('--lr_image_path', '-b', type=str, default="../data/Set14/LR/X4/monarch.png", help="LR image path") - self.parser.add_argument('--output_path', type=str, default="../models/", help="Project path.") + self.parser.add_argument('--output_path', type=str, default="../models/", help="Output path.") self.parser.add_argument('--batch_size', type=int, default=2**10, help="Size of the batch for the input domain values.") self.parser.add_argument('--mirror', action='store_true', default=False)