【亲测免费】 RigidMask 项目使用教程

RigidMask 项目使用教程

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

RigidMask 项目的目录结构如下：

rigidmask/
├── eval/
│   ├── eval_seg.py
│   ├── generate_visual.py
│   └── ...
├── submission.py
├── README.md
├── requirements.txt
├── config/
│   ├── default_config.yaml
│   └── ...
├── data/
│   ├── imgs/
│   │   ├── coral/
│   │   └── kitti/
│   └── ...
├── weights/
│   ├── rigidmask-sf/
│   │   └── weights.pth
│   └── rigidmask-kitti/
│       └── weights.pth
└── ...

目录结构介绍

• eval/: 包含评估和生成可视化结果的脚本。
• submission.py: 项目的启动文件。
• README.md: 项目说明文档。
• requirements.txt: 项目依赖文件。
• config/: 包含项目的配置文件。
• data/: 包含项目使用的数据集。
• weights/: 包含预训练的模型权重文件。

2. 项目的启动文件介绍

项目的启动文件是 submission.py，该文件用于运行和测试模型。以下是启动文件的基本使用方法：

CUDA_VISIBLE_DEVICES=1 python submission.py --dataset seq-coral --datapath input/imgs/coral/ --outdir /weights/$modelname/ --loadmodel /weights/$modelname/weights.pth --testres 1

启动文件参数介绍

• --dataset: 指定数据集类型，如 seq-coral 或 seq-kitti。
• --datapath: 指定输入数据的路径。
• --outdir: 指定输出结果的目录。
• --loadmodel: 指定预训练模型的路径。
• --testres: 指定测试分辨率。

3. 项目的配置文件介绍

项目的配置文件位于 config/ 目录下，默认配置文件为 default_config.yaml。以下是配置文件的基本结构：

model:
  name: rigidmask-sf
  weights_path: /weights/rigidmask-sf/weights.pth
dataset:
  type: seq-coral
  datapath: input/imgs/coral/
output:
  outdir: /weights/rigidmask-sf/
test:
  resolution: 1

配置文件参数介绍

• model: 模型相关配置。
  • name: 模型名称。
  • weights_path: 预训练模型权重路径。
• dataset: 数据集相关配置。
  • type: 数据集类型。
  • datapath: 数据集路径。
• output: 输出相关配置。
  • outdir: 输出结果目录。
• test: 测试相关配置。
  • resolution: 测试分辨率。

以上是 RigidMask 项目的基本使用教程，包括项目的目录结构、启动文件和配置文件的介绍。希望对您有所帮助！