浅析mmdetection在windows10系统环境中搭建过程(2)

日期：2020-05-12 栏目：破解天下浏览：次

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install .

会自动安装Cython

安装pycocotools：

是mmdection读取coco数据集必备包

在命令行输入

pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

安装mmdetection：

官方网站：https://github.com/open-mmlab/mmdetection

补充安装依赖包requirements：

pip install -r requirements.txt

修改编译文件：

1.将~Lib\site-packages\torch\utils\cpp_extension.py 中 info = info.decode().lower()修为 info=info.decode("utf8","ignore").lower()

最好将所有decode() 都改成decode("utf8","ignore")

修改setup.py文件中extra_compile_args 相关代码：

def make_cuda_ext(name, module, sources):
 
 return CUDAExtension(
 name='{}.{}'.format(module, name),
 sources=[os.path.join(*module.split('.'), p) for p in sources],
 extra_compile_args={
  'cxx': ["-DMS_WIN64","-MD"],
  'nvcc': [
  '-D__CUDA_NO_HALF_OPERATORS__',
  '-D__CUDA_NO_HALF_CONVERSIONS__',
  '-D__CUDA_NO_HALF2_OPERATORS__',
  ]
 })

2.当前文件夹运行

python setup.py build_ext --inplace 进行编译
python setup.py install develop 完成安装

运行pip list 查看是否安装成功，成功的话会有mmdet包以及对应版本

后续问题：

修改~\Lib\site-packages\mmcv\utils中的config.py文件中的123行的

with open(filename, 'r', encoding='utf-8') as f:

增加encoding='utf-8'

到此，在windows10中就可以正常运行mmdetecion框架代码了，之前有遇到focal loss不能正常使用，但是后来安装都是可以的，后续有什么问题欢迎大家提出，谢谢。

2020-01-10更新：

mmdetection在2019年12月13号进行了新版本的更新，其中对api/train.py增加torch.distributed，这块在windows下不支持，所以要在windows中训练的话需要把v1.0rc1的版本的train与新版本的train进行合并，主要是去除torch.distributed以及_non_dist_train的修改为主。

总结

以上所述是小编给大家介绍的mmdetection在windows10系统环境中搭建过程，希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对黑区网络网站的支持！

转载注明出处：http://www.heiqu.com/1634.html

浅析mmdetection在windows10系统环境中搭建过程(2)

相关推荐