AI-TOD数据集

官方获取途径:jwwangchn/AI-TOD: Official code for "Tiny Object Detection in Aerial Images". (github.com)

如果嫌麻烦,这里有转换好的数据集:极市开发者平台-计算机视觉算法开发落地平台-极市科技

AI-TOD和AI-TOD-V2两个数据集的图片都是一样的,但标签有差别,标签文件可以直接在GitHub上下载。

aitodpycocotools安装

GitHub链接:jwwangchn/cocoapi-aitod: COCO API - Dataset @ http://cocodataset.org/ (github.com)

项目里readme文件上提供的安装方式会受到网速的影响,可以将项目的zip文件下载下来,然后解压,进入到这个项目的目录中,执行以下代码

cd lvis
pip install -r requirements.txt
python setup.py install

 这两行指令运行结束后,终端输入如下日志

 这时可以看到aitodpycocotools文件夹下也多出来一个setup.py的脚本文件,执行以下指令

cd ../aitodpycocotools #进入到aitodpycocotools下
python setup.py install

 不出意外,就能完成安装。我在这一步出现了报错,供参考,报错如下:

gcc: error: ../common/maskApi.c: 没有那个文件或目录
gcc: fatal error: no input files
compilation terminated.
error: command '/usr/local/cuda-11.8/bin/gcc' failed with exit code 1

解决办法:降低cython的版本,本人将cython版本从3.0.11换成0.29.36,再次安装就成功了。

 

 

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐