charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

make model required

Browse Source
This commit is contained in:
lorenzo 2019-07-26 12:09:29 +02:00
parent cec1ba2c83
commit 698dbbd7be
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monoloco/run.py
View File

@ -43,10 +43,10 @@ def cli():
predict_parser.add_argument('--scale', default=1.0, type=float, help='change the scale of the image to preprocess')
# Monoloco
predict_parser.add_argument('--model', help='path of MonoLoco model to load',
default="data/models/monoloco-190719-0923.pkl")
predict_parser.add_argument('--model', help='path of MonoLoco model to load', required=True)
predict_parser.add_argument('--hidden_size', type=int, help='Number of hidden units in the model', default=512)
predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization', required=True)
predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization',
default='data/arrays/names-kitti-190513-1754.json')
predict_parser.add_argument('--transform', help='transformation for the pose', default='None')
predict_parser.add_argument('--draw_box', help='to draw box in the images', action='store_true')
predict_parser.add_argument('--predict', help='whether to make prediction', action='store_true')