交互式多类别微小物体检测项目教程
1. 项目目录结构及介绍
Interactive-Multi-Class-Tiny-Object-Detection/
├── configs/
│ ├── _base_/
│ │ ├── datasets/
│ │ ├── models/
│ │ ├── schedules/
│ │ └── default_runtime.py
│ ├── custom_config.py
│ └── README.md
├── datasets/
│ ├── Tiny-DOTA/
│ └── LCell/
├── demo/
│ ├── demo.py
│ └── README.md
├── tools/
│ ├── train.py
│ ├── test.py
│ └── README.md
├── README.md
└── requirements.txt
configs/
: 包含项目的配置文件,包括基础配置和自定义配置。datasets/
: 包含项目使用的数据集,如 Tiny-DOTA 和 LCell。demo/
: 包含演示脚本和说明。tools/
: 包含训练和测试脚本。README.md
: 项目的主说明文件。requirements.txt
: 项目依赖的 Python 包列表。
2. 项目的启动文件介绍
项目的启动文件主要位于 tools/
目录下:
train.py
: 用于启动训练过程的脚本。test.py
: 用于启动测试过程的脚本。
使用示例:
python tools/train.py --config configs/custom_config.py
python tools/test.py --config configs/custom_config.py
3. 项目的配置文件介绍
配置文件主要位于 configs/
目录下:
_base_/
: 包含基础配置文件,如数据集配置、模型配置、训练计划和默认运行时配置。custom_config.py
: 自定义配置文件,可以根据需要修改以适应不同的训练和测试需求。
配置文件示例:
# custom_config.py
_base_ = [
'../_base_/datasets/tiny_dota.py',
'../_base_/models/faster_rcnn_r50_fpn.py',
'../_base_/schedules/schedule_1x.py',
'../_base_/default_runtime.py'
]
# 自定义参数
data = dict(
samples_per_gpu=2,
workers_per_gpu=2,
)
通过修改 custom_config.py
文件中的参数,可以调整训练和测试的设置,如批量大小、工作线程数等。