monodepth

原始项目链接

https://github.com/mrharicot/monodepth

精度测试结果

代码和数据集

本次迁移选择KITTI数据集

【特殊说明】按照github源码方式下载数据集比较困难，可以直接到obs桶里面获取已经处理好的数据集。

获取数据集：obs://cann-id2099/dataset/KITTI/

快速开始

1. 目录说明

   数据集目录 /home/disk/xjk/dataset/KITTI/

   代码目录 ~/xjk/monodepth

2. 训练

   python3.7 ./monodepth_main.py --mode train --model_name my_model --filenames_file ./utils/filenames/kitti_train_files.txt --data_path /home/disk/xjk/dataset/KITTI/ --log_directory ~/tmp/
   

3. 测试

   python3.7 monodepth_main.py --mode test --data_path /home/disk/xjk/dataset/KITTI/ --filenames_file ./utils/filenames/kitti_stereo_2015_test_files.txt --log_directory ~/tmp/ --checkpoint_path ~/tmp/my_model/model-181250
   

4. 获取精度

   python3.7 ./utils/evaluate_kitti.py --split kitti --predicted_disp_path ~/tmp/my_model/disparities.npy --gt_path /home/disk/xjk/dataset/KITTI/