From 52ced2e75777d1e0966af1c4c0a0909823a3aa19 Mon Sep 17 00:00:00 2001 From: qslia Date: Sun, 10 Apr 2022 07:54:48 +0800 Subject: [PATCH 1/2] print assert error correctly Former-commit-id: 8c7feff7cf740e788228839883a592c87097add2 --- utils/data_loading.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/data_loading.py b/utils/data_loading.py index 63c4313..e4625ac 100644 --- a/utils/data_loading.py +++ b/utils/data_loading.py @@ -64,7 +64,7 @@ class BasicDataset(Dataset): img = self.load(img_file[0]) assert img.size == mask.size, \ - 'Image and mask {name} should be the same size, but are {img.size} and {mask.size}' + f'Image and mask {name} should be the same size, but are {img.size} and {mask.size}' img = self.preprocess(img, self.scale, is_mask=False) mask = self.preprocess(mask, self.scale, is_mask=True) From 109cf3462b02a7cd4aebda1ee56febf7604eb71e Mon Sep 17 00:00:00 2001 From: qslia Date: Sun, 10 Apr 2022 11:41:35 +0800 Subject: [PATCH 2/2] change args name Former-commit-id: 22020bdda52ed24e2163083dc1fa6889ba266225 --- predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict.py b/predict.py index 9cfe187..20a75e4 100755 --- a/predict.py +++ b/predict.py @@ -49,7 +49,7 @@ def get_args(): parser.add_argument('--model', '-m', default='MODEL.pth', metavar='FILE', help='Specify the file in which the model is stored') parser.add_argument('--input', '-i', metavar='INPUT', nargs='+', help='Filenames of input images', required=True) - parser.add_argument('--output', '-o', metavar='INPUT', nargs='+', help='Filenames of output images') + parser.add_argument('--output', '-o', metavar='OUTPUT', nargs='+', help='Filenames of output images') parser.add_argument('--viz', '-v', action='store_true', help='Visualize the images as they are processed') parser.add_argument('--no-save', '-n', action='store_true', help='Do not save the output masks')