PlanarRecon 项目使用教程

PlanarRecon 项目使用教程

PlanarRecon 项目地址: https://gitcode.com/gh_mirrors/pl/PlanarRecon

1. 项目目录结构及介绍

PlanarRecon 项目的目录结构如下：

PlanarRecon/
├── checkpoints/
│   └── release/
├── config/
│   ├── train_phase1.yaml
│   └── train_phase2.yaml
├── data/
│   └── scannet/
│       ├── scans/
│       └── scans_raw/
├── main.py
├── README.md
├── requirements.txt
└── setup.py

目录结构介绍

• checkpoints/: 存放预训练模型的权重文件。

  • release/: 存放发布版本的预训练模型。

• config/: 存放项目的配置文件。

  • train_phase1.yaml: 训练阶段1的配置文件。
  • train_phase2.yaml: 训练阶段2的配置文件。

• data/: 存放数据集文件。

  • scannet/: ScanNet 数据集的存放目录。
    • scans/: 存放 ScanNet 数据集的扫描数据。
    • scans_raw/: 存放 ScanNet 数据集的原始数据。

• main.py: 项目的启动文件。

• README.md: 项目的说明文档。

• requirements.txt: 项目的依赖库列表。

• setup.py: 项目的安装脚本。

2. 项目的启动文件介绍

项目的启动文件是 main.py。该文件负责启动 PlanarRecon 项目的主要功能，包括训练和测试。

启动命令示例

python main.py --cfg /config/train_phase1.yaml

参数说明

• --cfg: 指定配置文件路径，用于加载训练或测试的配置参数。

3. 项目的配置文件介绍

项目的配置文件存放在 config/ 目录下，主要包括 train_phase1.yaml 和 train_phase2.yaml。

配置文件内容示例

# train_phase1.yaml
MODEL:
  FUSION:
    FUSION_ON: True
  TRACKING: False

# train_phase2.yaml
MODEL:
  FUSION:
    FUSION_ON: True
  TRACKING: True

配置项说明

• MODEL: 模型相关的配置项。
  • FUSION: 融合相关的配置项。
    • FUSION_ON: 是否启用融合功能，True 表示启用，False 表示禁用。
  • TRACKING: 是否启用跟踪功能，True 表示启用，False 表示禁用。

通过修改这些配置文件，可以调整项目的训练和测试行为。

PlanarRecon 项目地址: https://gitcode.com/gh_mirrors/pl/PlanarRecon