PyTorch 加载 COCO 数据集需要安装 COCO API, COCO 对 Windows 是不支持的,不过为了支持 Windows ,有人对 COCO 做了一些修改: https://github.com/philferriere/cocoapi。

方法一:

(1)在  Github下载源码并进行解压;
(2)安装 Visual C++ ;(网上都说装2015或者2014版本的,我就直接装的,没有特意考虑特定的版本,最后也运行出来了)
(3)进入coco源码setup.py所在目录 cocoapi-master\PythonAPI;
(4)执行命令:python setup.py build_ext --inplace

(5)把编译好的pycocotools放到自己创建的虚拟环境下面的Lib\site-packages当中就可以。

(6)验证是否装成功。

在pycharm的python console中运行,看“from pycocotools.coco import COCO”是否导入成功。

方法一能不能安装成功全靠运气,

方法二:直接运行pip install pycocotools-windows。(首推)

编译后的文件夹

验证

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐