AI-TOD:空中图像中的微小目标检测开源项目安装教程
2026-01-25 05:36:53作者:宣利权Counsellor
1. 项目介绍
AI-TOD(论文)是专为空中图像设计的一个微小目标检测数据集,它包含在28,036张航拍图片中总计700,621个对象实例,跨越八大类别。相比于其他空基物体检测数据集,AI-TOD中的对象平均尺寸约为12.8像素,远小于一般标准。这个项目旨在帮助研究者应对极端条件下小目标的识别挑战。
2. 项目下载位置
您可以通过访问其GitHub仓库来获取项目源码:
[GitHub Repository](https://github.com/jwwangchn/AI-TOD)
直接点击页面上的“Code”按钮选择克隆或下载ZIP文件。
3. 安装环境配置
系统需求:
- 操作系统: Linux或macOS (Windows理论上可行但未明确支持)
- Python版本: 3.7
- 必要的库包括
mmcv,WWTool以及其他基础Python包。
图片示例步骤:
由于配置过程涉及命令行操作,这里不直接提供图片示例,但步骤清晰如下:
-
安装Python 3.7。
-
创建虚拟环境(可选,推荐):
python3.7 -m venv myenv source myenv/bin/activate -
安装MMCV:
pip install mmcv-full -
克隆WWTool并安装:
git clone https://github.com/jwwangchn/WWTool.git cd WWTool python setup.py develop cd .. -
克隆AI-TOD项目:
git clone https://github.com/jwwangchn/AI-TOD.git -
安装AI-TOD依赖: 进入到AI-TOD目录下并安装依赖:
cd AI-TOD pip install -r requirements.txt
4. 项目安装方式
AI-TOD项目不仅仅是代码库,还包括一个数据集合成工具。你需要下载两部分数据,并通过提供的脚本生成完整数据集。
-
组织下载的文件结构如下:
├── AI-TOD ├── annotations # 放置AI-TOD_wo_xview的标注文件(json) ├── images # 解压后的AI-TOD_wo_xview图像(png),按folder放置 ├── ... ├── xview # 放置xView数据 ├── ori └── train_images └── xView_train.geojson ├── generate_aitod_imgs.py # 合成脚本 -
运行合成脚本来生成AI-TOD数据集:
python generate_aitod_imgs.py
这一步可能需要大约一个小时的时间。
5. 项目处理脚本
主要处理脚本为generate_aitod_imgs.py,用于结合上述准备的数据,通过端到端的方式生成AI-TOD的最终图像数据集。确保您的环境已经正确配置,并且所有必要的数据都已经被妥善安放,之后执行此脚本即可完成数据集的生成。
以上就是AI-TOD项目从下载到初步运行的全过程。请注意,在进行模型训练或者实验之前,还应详细阅读项目文档和论文以理解数据集的具体使用方法以及模型的训练细节。
登录后查看全文
热门项目推荐
相关项目推荐
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0172
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook097
Step-3.7-FlashStep-3.7-Flash是一个拥有 1980 亿参数的稀疏混合专家(MoE)视觉语言模型,由 1960 亿参数的语言主干网络和 18 亿参数的视觉编码器组合而成,具备原生图像理解能力。Python00
BitCPM-CANN-8BBitCPM-CANN 是首个基于华为昇腾 NPU 原生构建的端到端 1.58 位(三值化)大语言模型训练系统。该系统将量化感知训练(QAT)集成到 Megatron-LM 框架中,并结合 MindSpeed 加速,覆盖了从自定义三值算子到基于昇腾 910B 的分布式并行训练的完整训练栈。Python00
MiniCPM5-1BMiniCPM5-1B,这是 MiniCPM5 系列的首款模型。它是一个专为端侧、本地部署和资源受限场景打造的 10 亿参数密集型 Transformer 模型,达到了 10 亿参数级开源模型的 SOTA 水平Jinja00
skillhubopenJiuwen 生态的 Skill 托管与分发开源方案,支持自建与可选 ClawHub 兼容。Python0239
热门内容推荐
项目优选
收起
deepin linux kernel
C
32
16
暂无描述
Dockerfile
750
4.87 K
Claude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed.
Get Started
Rust
1.58 K
172
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
841
1.84 K
Ascend Extension for PyTorch
Python
689
834
CANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。
Jupyter Notebook
229
97
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
451
418
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.02 K
1.04 K
暂无简介
Dart
999
259
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
642
1.27 K