TTFNet 开源项目使用教程
ttfnet项目地址:https://gitcode.com/gh_mirrors/tt/ttfnet
1. 项目的目录结构及介绍
TTFNet 项目的目录结构如下:
ttfnet/
├── configs/
│ └── ttfnet/
│ ├── ttfnet_d53_1x.py
│ ├── ttfnet_d53_2x.py
│ ├── ttfnet_r18_1x.py
│ ├── ttfnet_r18_2x.py
│ └── ...
├── demo/
├── docker/
├── docs/
├── img/
├── mmdet/
├── tests/
├── tools/
│ ├── dist_test.sh
│ ├── dist_train.sh
│ └── ...
├── .gitignore
├── .isort.cfg
├── .pre-commit-config.yaml
├── .style.yapf
├── .travis.yml
├── LICENSE
├── MMDETECTION_README.md
├── README.md
├── pytest.ini
├── requirements.txt
└── setup.py
目录结构介绍
configs/
: 包含项目的配置文件,如ttfnet_d53_1x.py
和ttfnet_d53_2x.py
等。demo/
: 包含演示代码和示例。docker/
: 包含 Docker 相关文件。docs/
: 包含项目文档。img/
: 包含项目所需的图像文件。mmdet/
: 包含 mmdetection 框架的核心代码。tests/
: 包含测试代码。tools/
: 包含训练和测试的脚本,如dist_test.sh
和dist_train.sh
。.gitignore
: Git 忽略文件。.isort.cfg
: isort 配置文件。.pre-commit-config.yaml
: pre-commit 配置文件。.style.yapf
: yapf 样式配置文件。.travis.yml
: Travis CI 配置文件。LICENSE
: 项目许可证。MMDETECTION_README.md
: mmdetection 的 README 文件。README.md
: 项目的主 README 文件。pytest.ini
: pytest 配置文件。requirements.txt
: 项目依赖文件。setup.py
: 项目安装脚本。
2. 项目的启动文件介绍
训练脚本
训练 TTFNet 模型可以使用 tools/dist_train.sh
脚本。以下是训练 TTFNet-18 和 TTFNet-53 的示例命令:
# 训练 TTFNet-18
./tools/dist_train.sh configs/ttfnet/ttfnet_r18_2x.py 8
# 训练 TTFNet-53
./tools/dist_train.sh configs/ttfnet/ttfnet_d53_1x.py 8
测试脚本
测试 TTFNet 模型可以使用 tools/dist_test.sh
脚本。以下是测试 TTFNet-53 的示例命令:
./tools/dist_test.sh configs/ttfnet/ttfnet_d53_2x.py /path/to/the/checkpoint 8
3. 项目的配置文件介绍
配置文件结构
配置文件位于 configs/ttfnet/
目录下,常见的配置文件包括:
ttfnet_d53_1x.py
ttfnet_d53_2x.py
ttfnet_r18_1x.py
ttfnet_r18_2x.py
配置文件示例
以下是一个简化的配置文件示例 ttfnet_d53_1x.py
:
# model settings
model = dict(
type='TTFNet',
backbone=dict(
type='Darknet',
depth=53,
out_indices=(3, 4, 5),
),
neck=None,
bbox_head=dict(
type='TTFHead',
in_channels=[1024, 512,