Skip to content

Commit cdba31f

Browse files
author
Clément Pinard
committed
make run_inference help clearer
1 parent 9bd4b54 commit cdba31f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

run_inference.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
parser.add_argument('pretrained', metavar='PTH', help='path to pre-trained model')
2525
parser.add_argument('--output', '-o', metavar='DIR', default=None,
2626
help='path to output folder. If not set, will be created in data folder')
27-
parser.add_argument('--output-value', '-v', metavar='VAL', choices=['raw', 'vis', 'both'], default='both',
27+
parser.add_argument('--output-value', '-v', choices=['raw', 'vis', 'both'], default='both',
2828
help='which value to output, between raw input (as a npy file) and color vizualisation (as an image file).'
2929
' If not set, will output both')
3030
parser.add_argument('--div-flow', default=20, type=float,
@@ -60,7 +60,6 @@ def main():
6060
save_path = Path(args.output)
6161
print('=> will save everything to {}'.format(save_path))
6262
save_path.makedirs_p()
63-
6463
# Data loading code
6564
input_transform = transforms.Compose([
6665
flow_transforms.ArrayToTensor(),

0 commit comments

Comments
 (0)