monodepth2跑代码笔记

Stereo training

Our code defaults to using Zhou's subsampled Eigen training data. For stereo-only training we have to specify that we want to use the full Eigen training set – see paper for details.

运行命令：

原文中没有数据路径（放在根目录下），这里我把数据集放在了其他地方，加上数据路径运行即可。

python train.py --model_name stereo_model \
  --frame_ids 0 --use_stereo --split eigen_full \
  --data_path  /home/csy_0418/dataset/kitti_data

遇到的问题​​​​

1.​​ModuleNotFoundError: No module named 'IPython'​​

2.跑完后训练好后的模型在/home/csy_0418下，这与选项--log_dir有关

self.parser.add_argument("--log_dir",
                         type=str,
                         help="log directory",
                         default=os.path.join(os.path.expanduser("~"), "tmp"))

所以训练的时候，可以设置该选项--log_dir='./tmp'

"./"：代表目前所在的目录。

成功运行！

python train.py

train.py代码学习与解读

整体很简单：

1.类的知识，创建实体options，获得参数。​​argparse模块学习。​​

2.关于__main__------还不清楚，问题不大，以后来解决

3.创建对象trainer，传入参数-----进行初始化

4.对trainer对象进行train（）方法

下面仔细看Trainer类

Trainer类仔细解读学习

整体框架

__init__

同distdepth差不多；之后仔细来看，对比来看；

train方法

trainer.train()

即运行该段代码。