DetNAS 项目使用教程

DetNAS 项目使用教程

DetNAS 项目地址: https://gitcode.com/gh_mirrors/de/DetNAS

1. 项目目录结构及介绍

DetNAS 项目的目录结构如下：

DetNAS/
├── configs/
│   ├── demo/
│   ├── distributed_arch_search/
│   ├── maskrcnn_benchmark/
│   └── tests/
├── docker/
├── scripts/
├── tools/
├── ABSTRACTIONS.md
├── CODE_OF_CONDUCT.md
├── CONTRIBUTING.md
├── INSTALL.md
├── LICENSE
├── MODEL_ZOO.md
├── README.md
├── TROUBLESHOOTING.md
├── config.sh
├── requirements.txt
└── setup.py

目录介绍

• configs/: 包含项目的配置文件和示例配置。

  • demo/: 包含演示配置文件。
  • distributed_arch_search/: 包含分布式架构搜索的配置文件。
  • maskrcnn_benchmark/: 包含基于 maskrcnn-benchmark 的配置文件。
  • tests/: 包含测试配置文件。

• docker/: 包含 Docker 相关的文件和配置。

• scripts/: 包含项目的启动脚本和训练脚本。

• tools/: 包含项目使用的工具和辅助脚本。

• ABSTRACTIONS.md: 抽象概念的说明文档。

• CODE_OF_CONDUCT.md: 项目的行为准则。

• CONTRIBUTING.md: 贡献指南。

• INSTALL.md: 安装指南。

• LICENSE: 项目的开源许可证。

• MODEL_ZOO.md: 模型库的说明文档。

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

• TROUBLESHOOTING.md: 故障排除指南。

• config.sh: 项目的配置脚本。

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

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

2. 项目启动文件介绍

DetNAS 项目的启动文件主要集中在 scripts/ 目录下。以下是一些关键的启动脚本：

• scripts/run_detnas_coco_fpn_300M.sh: 用于启动 COCO 数据集上的 FPN 模型训练。
• scripts/run_detnas_coco_fpn_300M_search.sh: 用于启动 COCO 数据集上的 FPN 模型搜索。
• scripts/run_server.sh: 用于启动分布式搜索的服务器。

启动示例

bash scripts/run_detnas_coco_fpn_300M.sh

3. 项目的配置文件介绍

DetNAS 项目的配置文件主要集中在 configs/ 目录下。以下是一些关键的配置文件：

• configs/maskrcnn_benchmark/config.py: 包含 maskrcnn-benchmark 的主要配置。
• configs/distributed_arch_search/arch_search_config.py: 包含分布式架构搜索的配置。
• config.sh: 项目的全局配置脚本，用于设置数据集路径等。

配置示例

在 config.sh 中，你可以修改数据集路径：

COCO_PATH=/path/to/your/coco/dataset

在 configs/maskrcnn_benchmark/config.py 中，你可以修改模型的超参数：

MODEL = dict(
    BACKBONE = dict(
        CONV_BODY = 'FPN-300M',
        FREEZE_CONV_BODY_AT = 2,
    ),
    RESNETS = dict(
        NUM_GROUPS = 1,
        WIDTH_PER_GROUP = 64,
        STRIDE_IN_1X1 = False,
    ),
)

通过以上配置文件，你可以根据需要调整项目的运行参数和数据集路径。

DetNAS 项目地址: https://gitcode.com/gh_mirrors/de/DetNAS