swin transformer obect detection复现(CPU+GPU)

1. 代码下载+权重下载（我是在主目录下新建了checkpoints文件下，然后把权重放进去的）
2. conda create –n st python=3.8
3. source activate st
4. pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2
5. python setup.py install
6. pip install openmim
7. mim install mmcv-full==1.4.0
8. pip install -r requirements.txt
9. python setup.py develop
10. python demo/image_demo.py demo/demo.jpg configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py ./checkpoints/mask_rcnn_swin_tiny_patch4_window7_1x.pth
     

主要提示：

1、第三步安装时注意torch和torchvision的版本对应问题，官网可以查

2、第六步安装时一定要指定版本，我自己安装时默认安装mmcv1.4.5的版本，虽然作者GitHub官网写可以用1.4.5版本，但是安装好之后会报错这个

 查阅相关博客修改代码之后发现还是有问题，会有这样的问题，所以干脆就直接安装1.4.0版本的mmcv了

3、没有GPU的时候用CPU推理图片时记得后面加参数 --device cpu

4、还有提醒大家下载master版本代码，不要下载分支版本，我已经踩过坑了

结束 当然中间还遇到了好多问题，但是都没记录下来，发布的时安装这个流程下来在我的电脑上可以安装好的。记录一下，防止今后又要配环境，还有我没有安装apex，听说那个雷也挺多的，我担心出是什么问题又要重新配环境，不过apex好像是GPU混合精度训练时用的，我暂时用不到，今后我用到了还会更新，有什么问题大家可以交流

-------------------------------------------------------------------------------------------------------------------------

我已经在公司服务器上用GPU跑通了，所以也可以用GPU版本的了，记录一下：

1、先卸载之前安装的mmcv-full，pip uninstall mmvc-full，然后登陆

登录https://github.com/open-mmlab/mmcv/blob/master/README_zh-CN.md，选择适合自己电脑安装的torch和cuda版本

点击安装后会出现一个pip命令：

pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html

 把中间测mmcv_version换成自己想安装的版本，我用的是1.4.0 ，太大的版本我跑不起来。

一般要是安装好应该就可以运行了，但是我在服务器上还遇到了一个错误：importerror：libcudart.so.11.0: cannot open share object file: No such file or directory。解决办法：在终端添加动态库的指令：export LD_LIBRARY_PATH={自己的cuda安装路径/lib/:}$LD_LIBRARY_PATH

添加这个动态库就可以正常运行啦。

最后

以上就是坚强老鼠为你收集整理的swin transformer obect detection复现(CPU+GPU)的全部内容，希望文章能够帮你解决swin transformer obect detection复现(CPU+GPU)所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。