解决mmdetection显示mask的问题

最近mmdetection更新了2.0版本,更新了一点api。写一下自己的笔记吧
官方给出的测试demo是使用faster-rcnn模型,而我在使用别的模型时,尝试想画出mask的时候一直报错,如下:

Traceback (most recent call last):
  File "demo.py", line 21, in <module>
    show_result_pyplot(model, img, result, score_thr=0.3)
  File "/mmdetection/mmdet/apis/inference.py", line 146, in show_result_pyplot
    img = model.show_result(img, result, score_thr=score_thr, show=False)
  File "/mmdetection/mmdet/models/detectors/base.py", line 211, in show_result
    mask = maskUtils.decode(segms[i]).astype(np.bool)
  File 
  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
mmdetection swin maskrcnn是一个基于Swin Transformer的目标检测模型。你可以使用命令"python demo/image_demo.py demo/52.tif configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py work_dirs/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco/latest.pth"来进行测试。\[1\] 在配置文件中,你可以修改configs文件夹下的相关配置,例如albu_example、atss和_base_等文件夹。其中_base_文件夹是最根本的继承,包含了数据集的训练方法、数据增强、测试方法等。\[2\] 如果你使用的是自定义数据集,需要注意是否标注了mask。通常情况下,自定义数据集只会标注bbox,所以需要禁用mask-rcnn中的mask。具体操作包括: 1. 在基础模型文件cascade_mask_rcnn_swin_fpn.py中将use_mask由True改为False,并注释掉mask_roi_extractor和mask_head两个变量。 2. 在入口配置文件cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py中,将train_pipeline中的with_mask由True改为False,并在dict(type='Collect', keys=\['img', 'gt_bboxes', 'gt_labels', 'gt_masks'\])中去掉'gt_masks'。 3. 在数据集描述文件coco_instance.py中,将evaluation中的metric中去掉'segm'。\[3\] 这样配置之后,你就可以使用mmdetection swin maskrcnn进行目标检测了。 #### 引用[.reference_title] - *1* [Ubuntu和Windows使用Mmdetection训练Swin-Transformer+Mask-RCNN](https://blog.csdn.net/Wddym/article/details/125268262)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [利用mmdetection训练自己数据集过程error解决汇总](https://blog.csdn.net/QQ992281036/article/details/126979053)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [swing transformer中修改mmdetection预训练mask-rcnn使类别数适应custom dataset](https://blog.csdn.net/zkp_987/article/details/125724107)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值