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

add required fields

Browse Source
This commit is contained in:
lorenzo 2019-05-13 17:41:10 +02:00
parent 511b9ae650
commit 456bbc0620
2 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -45,7 +45,7 @@ Run the following to create the folders:
```
mkdir data
cd data
mkdir arrays models data-kitti data-nuscenes logs
mkdir arrays models kitti nuscenes logs
```
### Pre-trained Models
@ -126,13 +126,13 @@ nuScenes or KITTI Dataset. To create the joints run: `python src/main.py prep` s
dataset are supported: nuscenes_mini, nuscenes, nuscenes_teaser.
### Ground truth file for evaluation
The preprocessing script also outputs a second json file called **names.json** which provide a dictionary indexed
The preprocessing script also outputs a second json file called **names-<date-time>.json** which provide a dictionary indexed
by the image name to easily access ground truth files for evaluation and prediction purposes.
# Train
Provide the json file containing the preprocess joints (**joints.json**) as argument.
Provide the json file containing the preprocess joints as argument.
As simple as `python3 src/main.py --train --joints 'data/arrays/joints.json`

11
src/main.py
View File

@ -30,11 +30,10 @@ def cli():
eval_parser = subparsers.add_parser("eval")
# Preprocess input data
prep_parser.add_argument('--dir_ann', help='directory of annotations of 2d joints', required=True)
prep_parser.add_argument('--dataset',
help='datasets to preprocess: nuscenes, nuscenes_teaser, nuscenes_mini, kitti',
default='nuscenes')
prep_parser.add_argument('--dir_ann', help='directory of annotations of 2d joints',
default='/data/lorenzo-data/nuscenes_new/annotations/scale_1_april_new')
prep_parser.add_argument('--dir_nuscenes', help='directory of nuscenes devkit',
default='data/nuscenes/')
@ -60,7 +59,7 @@ def cli():
predict_parser.add_argument('--model', help='path of MonoLoco model to load',
default="data/models/best_model__seed_2_.pickle")
predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization',
default='data/arrays/names-kitti-000.json')
default='data/arrays/names-kitti-190513-1428.json')
predict_parser.add_argument('--dir_calib', default='data/kitti/calib/', help='directory of calib_files')
predict_parser.add_argument('--transform', help='transformation for the pose', default='None')
predict_parser.add_argument('--draw_kps', help='to draw kps in the images', action='store_true')
@ -91,9 +90,8 @@ def cli():
# Evaluation
eval_parser.add_argument('--dataset', help='datasets to evaluate, kitti or nuscenes', default='kitti')
eval_parser.add_argument('--geometric', help='to evaluate geometric distance', action='store_true')
eval_parser.add_argument('--dir_ann', help='directory of annotations of 2d joints',
default='/data/lorenzo-data/kitti/annotations/scale_2_april')
eval_parser.add_argument('--model', help='path of MonoLoco model to load')
eval_parser.add_argument('--dir_ann', help='directory of annotations of 2d joints', required=True)
eval_parser.add_argument('--model', help='path of MonoLoco model to load', required=True)
eval_parser.add_argument('--joints', help='Json file with input joints to evaluate (for nuscenes)',
default='data/arrays/joints-nuscenes_teaser-190513-1423.json')
eval_parser.add_argument('--n_dropout', type=int, help='Epistemic uncertainty evaluation', default=0)
@ -102,7 +100,6 @@ def cli():
eval_parser.add_argument('--hidden_size', type=int, help='Number of hidden units in the model', default=256)
eval_parser.add_argument('--n_stage', type=int, help='Number of stages in the model', default=3)
eval_parser.add_argument('--show', help='whether to show eval statistics', action='store_true')
eval_parser.add_argument('--trial', help='just to test', action='store_true')
args = parser.parse_args()
return args