-
Notifications
You must be signed in to change notification settings - Fork 0
/
How_to_run_code.txt
78 lines (72 loc) · 1.88 KB
/
How_to_run_code.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
# ################################################
====================================
Requirements
====================================
python 3.6
tensorflow 1.13.1
Ubuntu 16.04
cuda 10.0
====================================
dataset pretrain
====================================
# eigen
python data/prepare_train_data.py
--dataset_dir=/path/to/kitti/dataset/
--dataset_name=kitti_raw_eigen
--dump_root=/path/to/formatted/data/
--seq_length=2
--img_height=128
--img_width=416
--num_threads=16
--remove_static
# odom
python data/prepare_train_data.py
--dataset_dir=/path/to/kitti/dataset/
--dataset_name=kitti_odom
--dump_root=/path/to/formatted/data/
--seq_length=2
--img_height=128
--img_width=416
--num_threads=16
--remove_static
====================================
train mode
====================================
python mdvo_main.py
--mode=train_rigid
--dataset_dir=/path/to/formatted/data/
--checkpoint_dir=/path/to/save/ckpts/
--summary_out_dir=/path/to/save/ckpts/summary/
--learning_rate=0.0001
--seq_length=2
--batch_size=4
--max_steps=400100
--add_gan=True
====================================
mode test (pose)
====================================
python mdvo_main.py
--mode=test_pose
--dataset_dir=/path/to/kitti/odom/dataset/
--init_ckpt_file=/path/to/trained/model-xxx
--batch_size=1
--seq_length=2
--pose_test_seq=09
--output_dir=/path/to/save/predictions/result/
====================================
mode test (depth)
====================================
python mdvo_main.py
--mode=test_depth
--dataset_dir=/path/to/KITTI/data2015/
--init_ckpt_file=/path/to/trained/model-xxx
--batch_size=1
--depth_test_split=stereo
--output_dir=/path/to/save/predictions/result/
====================================
depth eval
====================================
python eval_depth.py
--kitti_dir=/path/to/KITTI/data2015/
--pred_file=/path/to/save/predictions/result/xx.npy
--output_dir=/path/to/save/predictions/result/