[踩坑记录]swin-Transformer的安装与调试

最近在用swin-Transformer跑CrowdHuman数据集,这个过程中遇到的各种坑记录一下。

  1. 安装的时候,不管swin-TRM的版本是什么,一定要在训练之前加上export PYTHONPATH=/path/to/yourdir,这是因为官方的swin-TRM包里面自带一个mmdet,你需要让mmdet的路径是你包里的,而不是你pip下载的;

  1. 安装apex的时候,要注意安装符合自己版本的,不能直接pip一下了事,我的就是apex-22.04-dev,去github下载后解压cd进去,然后pip install -r requirements.txt,把所有依赖包都下载了,再python setup.py install就可以了,不用网上那些复杂的步骤;

  1. mmcv-full,这是一个究极大坑,这得展开说一下!!!

  1. 首先,你需要确保你的pytorch版本和你GPU的算力能够互相匹配;在我这里,简要来说,就是3090你得装pytorch1.10.0,不能是1.8.0,别问我为什么,我也不理解;

  1. 确认匹配之后,根据CUDA版本和torch版本,在改变这个链接,找到对应网站:https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html,比如说我的cuda版本事11.3,torch版本事1.10.0,就写/cu113/torch1.10.0,进去之后,根据你需要根据你项目中的mmdet,注意!不是你pip install mmdet下载的mmdet的版本,来选择mmcv-full的版本(不知道是啥的话可以随机下一个,到时候报错会提醒你的,我这里是mmcv>=1.2.4, <=1.4.0),换句话说,pip install mmdet这句话就是一句没有用的话。然后再匹配python的版本,找到对应的whl文件下载。当然你得下载linux版本的(这里叫manylinux1,也不知道怎么起的名字),下载完直接去pip install /路径就可以啦;

  1. 在训练后验证的时候,可能会出现(KeyError(‘metric mAP is not supported‘.format(metric)))错误,这是由于你的验证集是coco格式,把--eval mAP改成--eval bbox就可以了

  1. 在这里补充一下我训练和验证的语句:

  1. 训练:tools/dist_train.sh configs/swin/cascade_mask_rcnn_swin_base_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py 3

  1. 验证:python tools/test.py configs/swin/cascade_mask_rcnn_swin_base_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py /root/autodl-twork_dirs/cp_base_3x/epoch_50.pth --show-dir detect_img --show --eval bbox

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Amigo_5610

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值