Skip to content

Commit 7ebd50c

Browse files
author
Clément Pinard
committed
convert options to float
1 parent 4c33f2a commit 7ebd50c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

run_inference.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,10 @@
2323
parser.add_argument('pretrained', metavar='PTH', help='path to pre-trained model')
2424
parser.add_argument('--output', metavar='DIR', default=None,
2525
help='path to output folder. If not set, will be created in data folder')
26-
parser.add_argument('--div-flow', default=20,
26+
parser.add_argument('--div-flow', default=20, type=float,
2727
help='value by which flow will be divided. overwritten if stored in pretrained file')
2828
parser.add_argument("--img-exts", default=['png', 'jpg', 'bmp'], nargs='*', type=str, help="images extensions to glob")
29-
parser.add_argument('--max_flow', default=None,
29+
parser.add_argument('--max_flow', default=None, type=float,
3030
help='max flow value. Flow map color is saturated above this value. If not set, will use flow map\'s max value')
3131

3232
best_EPE = -1

0 commit comments

Comments
 (0)