swin transformer obect detection复现(CPU+GPU)
Linux下配swin transformer目标检测环境
- 代码下载+权重下载(我是在主目录下新建了checkpoints文件下,然后把权重放进去的)
- conda create –n st python=3.8
- source activate st
- pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2
- python setup.py install
- pip install openmim
- mim install mmcv-full==1.4.0
- pip install -r requirements.txt
- python setup.py develop
- 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
添加这个动态库就可以正常运行啦。
更多推荐
所有评论(0)