产品简介
AI-TOD 是一个专为航空图像（无人机、卫星图）中小目标检测设计的基准数据集。它解决了现有数据集中物体尺寸过大、难以评估检测器在微小物体上性能的问题。主要面向计算机视觉领域的研究人员和开发者，用于训练和评估在复杂航拍场景下检测车辆、船舶等微小物体的AI模型。
核心功能
- 提供8类航拍常见目标的标准化数据集
- 包含超70万个物体实例，均值尺寸仅12.8像素
- 提供端到端数据集生成与合成脚本
- 配套专用评估工具(cocoapi-aitod)
- 数据集在学术场景下免费开放使用
快速开始
- 第一步：从GitHub克隆项目仓库，并下载xView训练集及AI-TOD_wo_xview部分。
- 第二步：按照仓库说明的目录结构，整理所有下载的图片和标注文件。
- 第三步：安装Python 3.7环境、mmcv及项目所需的依赖包(wwtool, requirements.txt)。
- 第四步：运行生成脚本`python generate_aitod_imgs.py`，约一小时后可在aitod文件夹获得完整数据集。
界面预览
与其他方案对比
| 对比项 | AI-TOD | 同类方案 |
|---|---|---|
| 物体平均尺寸 | 均值约12.8像素，专注评估微小物体检测能力 | 现有航拍数据集物体尺寸普遍较大，评估小目标能力不足 |
| 数据规模与类别 | 包含28,036张图像，8类目标，超70万个实例 | 同类数据集在物体尺寸、类别或规模上侧重点不同 |
| 工具完整性 | 提供从数据生成到模型评估的完整端到端工具链与脚本 | 部分数据集可能只提供原始数据，缺乏配套生成或评估工具 |
适合谁用
- 研究航空图像或遥感图像小目标检测算法的计算机视觉研究人员。
- 开发无人机巡检、卫星图像分析等应用的AI工程师。
- 致力于提升微小物体检测模型性能的算法研究者。
下载与版本
当前最新版本为 master，发布于 。支持 Windows、macOS、Linux。
⚠️ 注意：本项目仅提供源代码下载，不包含可直接运行的安装包（exe/dmg/apk 等。下载后需要自行编译构建才能使用，适合有一定开发经验的用户。如果你不熟悉编译流程，建议寻找同类开箱即用的替代方案。
注意事项
- 本工具仅供学习和研究使用，请遵守相关法律法规。
- 请尊重内容创作者版权，勿将下载内容用于商业或非法传播。
- 项目源码地址：https://github.com/jwwangchn/AI-TOD