AI-TOD 项目使用教程

AI-TOD 项目使用教程

AI-TODOfficial code for "Tiny Object Detection in Aerial Images".项目地址:https://gitcode.com/gh_mirrors/ai/AI-TOD

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

AI-TOD 项目的目录结构如下：

AI-TOD/
├── aitod/
│   ├── annotations/
│   ├── images/
│   │   ├── test/
│   │   ├── train/
│   │   ├── trainval/
│   │   └── val/
│   └── aitod_xview/
├── xview/
│   ├── ori/
│   │   └── train_images/
│   └── xView_train.geojson
├── generate_aitod_imgs.py
├── README.md
└── LICENSE

目录结构介绍

• aitod/: 包含 AI-TOD 数据集的标注文件和图像文件。
  • annotations/: 存放 AI-TOD 数据集的标注文件（JSON 格式）。
  • images/: 存放 AI-TOD 数据集的图像文件（PNG 格式）。
    • test/: 测试集图像。
    • train/: 训练集图像。
    • trainval/: 训练验证集图像。
    • val/: 验证集图像。
  • aitod_xview/: 包含与 xView 数据集相关的文件。
• xview/: 包含 xView 数据集的原始图像和标注文件。
  • ori/: 原始图像文件夹。
    • train_images/: 存放 xView 训练集的图像（TIF 格式）。
  • xView_train.geojson: xView 训练集的标注文件。
• generate_aitod_imgs.py: 用于生成完整 AI-TOD 数据集的端到端工具。
• README.md: 项目说明文档。
• LICENSE: 项目许可证文件。

2. 项目的启动文件介绍

项目的启动文件是 generate_aitod_imgs.py，这是一个端到端工具，用于生成完整的 AI-TOD 数据集。

启动文件介绍

• generate_aitod_imgs.py: 该脚本负责组织下载的文件并生成完整的 AI-TOD 数据集。运行该脚本可能需要大约一个小时，生成的完整图像集将位于 aitod 文件夹中。

3. 项目的配置文件介绍

AI-TOD 项目没有明确的配置文件，但可以通过以下方式进行配置和安装所需的包：

配置和安装

1. 安装环境：

   • Python 3.7
   • mmcv

2. 安装 wwtool：

   git clone https://github.com/jwwangchn/wwtool.git
   cd wwtool
   python setup.py develop
   

3. 安装其他所需包：

   cd aitodtoolkit
   pip install -r requirements.txt
   

4. 运行端到端工具：

   python generate_aitod_imgs.py
   

通过以上步骤，您可以配置和运行 AI-TOD 项目，生成完整的 AI-TOD 数据集。

AI-TODOfficial code for "Tiny Object Detection in Aerial Images".项目地址:https://gitcode.com/gh_mirrors/ai/AI-TOD