目标检测:swin-transformer跑通自己的数据集

环境:win11 单gpu
首先下载https://github.com/SwinTransformer/Swin-Transformer-Object-Detection

一开始安装总是报错,然后按照它的教程重新conda了一个新环境。
https://github.com/open-mmlab/mmdetection/blob/master/docs/en/get_started.md
step1,2,3,4…

记录一下碰到的问题

1.报错RuntimeError: Default process group has not been initialized, please make sure to call init_process_.
如果是多GPU训练,采用“SyncBN”; 如果是单GPU训练,将type修改为’BN’。
修改:原来是SyncBN,改成BN
在这里插入图片描述
2.深度学习Apex库出现报错‘IndexError: tuple index out of range‘的解决方法
if cached_x.grad_fn.next_functions[1][0].variable is not x: IndexError: tuple index out of range
参考的是https://blog.csdn.net/starvapour/article/details/122275439
在这里插入图片描述
比较好的解决问题的文章https://blog.csdn.net/CSDN_X_W/article/details/123845728

3.ModuleNotFoundError: No module named ‘mmcv._ext’
应该是mmcv和torch的版本的适配问题。我现在用的是mmcv1.5.2,它可以是会报错,说版本太大,只要修改一下它的最大版本号应该就能用了。
在这里插入图片描述
4.TypeError: init() got an unexpected keyword argument ‘embed_dim’
参考https://blog.csdn.net/xyzxyzxyz1999/article/details/123669850
我记得是把mmdet install好了,就没有问题了。

5.Apex is not installed
参考https://www.machunjie.com/linux/1075.html
在这里插入图片描述

训练自己的数据集
1.修改下面两个文件里面的类别名,分别是CLASSES,coco_classes()。
在这里插入图片描述
2.修改num_classes
在这里插入图片描述
我是用的这两个文件,所以修改这两个里面的内容

3.然后解决了碰到的问题,命令

 **python tools/train.py configs/swin/cascade_mask_rcnn_swin_base_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py**

开始训练了!
在这里插入图片描述

  • 1
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

骨子带刺

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

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

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

打赏作者

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

抵扣说明:

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

余额充值