mmdetection2环境搭建(windows10)

附上开源地址https://github.com/open-mmlab/mmdetection
如果能看得懂英文,尽量按照官方的来
首先需要创建一个新的虚拟环境,可以参考我之前的博客

安装pytorch,可以前往pytorch查看命令

conda install pytorch torchvision cudatoolkit=10.1 

进入下载好的mmdetection文件夹
安装必要的东西

1、修改D:\ProgramData\Anaconda3\envs\mmd2\Lib\site-packages\torch\utils\cpp_extension.py
185行:match = re.search(r’(\d+).(\d+).(\d+)’, compiler_info.decode(“utf8”,“ignore”).strip())
2、D:\ProgramData\Anaconda3\envs\mmd2\Lib\site-packages\torch\include\c10\util
替换 flat_hash_map.h
D:\ProgramData\Anaconda3\envs\mmd2\Lib\site-packages\torch\include\c10\util
替换 order_preserving_flat_hash_map.h
3、安装mmcv
cd mmcv-0.5.8
pip install -e .
4、安装pycocotools
cd cocoapi/PythonAPI
python setup.py build_ext install
5、进入mmdetection2.0
pip install -r requirements.txt
6、修改setup.py文件中CUDAExtension中extra_compile_args 相关代码,增加cxx的:"-DMS_WIN64","-MD"
7、python setup.py develop

测试代码

python demo/webcam_demo.py configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

参考博客:https://blog.csdn.net/trustguan/article/details/106537518

猜你喜欢

转载自blog.csdn.net/weixin_44868057/article/details/107868566
今日推荐