MonoPSR 开源项目使用教程-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_01029/article/details/142042371

MonoPSR 开源项目使用教程

monopsrMonocular 3D Object Detection项目地址:https://gitcode.com/gh_mirrors/mo/monopsr

1. 项目的目录结构及介绍

MonoPSR 项目的目录结构如下：

monopsr/
├── src/
│   ├── monopsr/
│   │   ├── configs/
│   │   ├── experiments/
│   │   ├── ...
│   ├── tf_ops/
│   │   ├── approxmatch/
│   │   ├── nn_distance/
│   │   ├── ...
├── requirements.txt
├── README.md
├── ...

目录结构介绍

src/: 包含项目的核心代码。
- monopsr/: 包含 MonoPSR 的主要功能模块。
  - configs/: 存放项目的配置文件。
  - experiments/: 存放实验相关的脚本和代码。
- tf_ops/: 包含 TensorFlow 的自定义操作。
  - approxmatch/: 包含用于近似匹配的自定义 TensorFlow 操作。
  - nn_distance/: 包含用于最近邻距离计算的自定义 TensorFlow 操作。
requirements.txt: 列出了项目所需的 Python 依赖包。
README.md: 项目的介绍文档。

2. 项目的启动文件介绍

MonoPSR 项目的启动文件主要位于 src/monopsr/experiments/ 目录下。主要的启动脚本是 run_inference.py。

`run_inference.py`

该脚本用于启动 MonoPSR 的推理过程。使用方法如下：

python src/monopsr/experiments/run_inference.py --config_path='src/monopsr/configs/monopsr_model_000.yaml' --default_ckpt_num='100000' --data_split='val'

参数说明

--config_path: 指定配置文件的路径。
--default_ckpt_num: 指定使用的模型检查点的编号。
--data_split: 指定数据集的分割（如 val 表示验证集）。

3. 项目的配置文件介绍

MonoPSR 的配置文件主要位于 src/monopsr/configs/ 目录下。主要的配置文件是 monopsr_model_000.yaml。

`monopsr_model_000.yaml`

该配置文件包含了 MonoPSR 模型的各项配置参数，如数据路径、模型参数、训练参数等。以下是配置文件的部分内容示例：

model:
  name: "monopsr_model_000"
  input_shape: [384, 1280, 3]
  ...

training:
  batch_size: 8
  learning_rate: 0.001
  ...

data:
  kitti_root_dir: "~/Kitti/object"
  ...