Win10系统下安装detectron2–精简版
前言
最近有要用到detectron2这个框架,网上介绍是一个非常好用的目标检测分割的开源项目,虽然不如mmdet支持的模型多,但好再灵活性高,可以满足造轮子的基本要求,所以,还有什么好说的呢~别人有的俺也要有!
环境准备
我用的是python3.8,torch1.8.1+cu111,需要先安装好一些基本的库及detectron2依赖的库,主要的几个包括pycocotools,fvcore,ninja,网上很多详细教程,本人也是一步一个坑蹚过来的,个中辛酸就不在这边细说了。对了,建议还是新建一个虚拟环境,防止接下来的操作会影响其他库的正常使用。
detectron2安装
1、首先,去github上下载代码包detectron2-master到本地并解压缩。
2、修改文件(这边很重要,否则很令人头大)
1)Anaconda3安装位置\envs\自己的虚拟环境\Lib\site-packages\torch\include\torch\csrc\jit\runtime\argument_spec.h 160行修改为:
static const size_t ARG_SPEC_DEPTH_LIMIT = 128;
2)Anaconda3安装位置\envs\自己的虚拟环境\Lib\site-packages\torch\include\pybind11\cast.h ,注意一下,旧版的里面说是在1449行里要修改explicit operator type&() { return *(this->value); },新版代码出现在1535行左右,不需要修改(我之前想改的,找不多1449行中的内容),啥也不动就可以。
3)Anaconda3安装位置\envs\自己的虚拟环境\Lib\site-packages\torch\utils\cpp_extension.py 299行修改为match = re.search(r’(\d+).(\d+).(\d+)‘, compiler_info.decode(’ gbk’).strip())
4)下载的detectron2\detectron2\layers\csrc\cocoeval\cocoeval.cpp修改487行为localtime_s(&local_time,&rawtime)
详细的detectron2安装的教程指路
安装教程1
安装教程2
最后在python里面import看一下则安装成功
仅供参考哈~
希望大家炼丹顺利!
溜了溜了,嘿嘿~