From 109cf3462b02a7cd4aebda1ee56febf7604eb71e Mon Sep 17 00:00:00 2001 From: qslia Date: Sun, 10 Apr 2022 11:41:35 +0800 Subject: [PATCH] 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')