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')