AI-TOD数据集获取及aitodpycocotools安装
·
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,再次安装就成功了。
更多推荐
所有评论(0)