monodepth2学习3-代码目录讲解

代码目录讲解

这里对个人对代码的理解进行讲解,个人由于设备不太行,没有去对模型进行复现,仅集合了网上内容对代码进行了解读,希望可以有帮助。

目录结构

  • asserts:这个主要是包含readme中需要使用的图片
  • datasets:这个文件内包含kitti_dataset.py和mono_dataset.py这部分主要是用于继承了Dataset类用于获取数据
  • experiments:主要是训练相关指令,方便查看
  • networks:主要是神经网络结构
  • splits:文件夹主要是KITTI数据集的下载相关文件目录,包括不同训练类型下使用到的训练文件,这里具体使用哪个可以查看readme
  • depth_prediction_example.ipynb:这个是用于验证的文件
  • evaluate_depth.py: 不参与到训练,主要是测试时进行验证使用,和验证深度网络有效性
  • evaluation_pose.py:用于确定相机姿势网络的有效性
  • export_gt_depth.py:不参与训练,大概用于深度kitti图像的深度信息
  • kitti_utils.py:读取kitti数据
  • layers:辅助函数,参与计算,主要是涉及训练过程中的一些图像转化函数,实现了计算损失、生成点云、生成二维图像、计算T和R等功能。
  • options: 具体的训练指令相关文件,用于使用命令行进行训练,对各个训练参数都进行了描述
  • text_simple.py: 验证文件
  • trainer.py:具体的训练过程
  • train.py:生成trainer对象,进行训练
  • utils.py:辅助函数

这里面比较重要的是network文件、trainer.py和layer.py,涉及到了整个训练过程,之后将对训练过程进行注释,方便各位理解。

posted @ 2022-08-20 11:01  black-world  阅读(807)  评论(0编辑  收藏  举报