【深度学习mmdetection错误】——mmdetection 运行报错KeyError:‘ConvWS is already registered in conv layer‘

于是修改以下mmdetection的安装文件:
site-packages/mmdet-2.1.0+unknown-py3.7-linux-x86_64.egg/mmdet/ops/conv_ws.py"

@CONV_LAYERS.register_module('ConvWS')

修改为:
@CONV_LAYERS.register_module(name='ConvWS', force=True)
 

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
这个错误通常是由于代码中的某些变量或方法未正确定义或导入而引起的。在这种情况下,错误信息表明在SSDHead对象中找不到loss_cls属性。这可能是由于以下原因之一导致的: 1.代码中确实没有定义loss_cls属性或方法。 2.代码中定义了loss_cls属性或方法,但是由于某些原因未正确导入或初始化。 3.代码中定义了loss_cls属性或方法,但是在SSDHead对象中未正确调用。 为了解决这个问题,你可以尝试以下几个步骤: 1.检查代码中是否正确定义了loss_cls属性或方法,并确保它们被正确导入和初始化。 2.检查代码中是否正确调用了loss_cls属性或方法,并确保它们被正确传递和使用。 3.检查代码中是否存在拼写错误或语法错误,并进行必要的更正。 4.检查代码中是否存在其他与此错误相关的警告或错误,并进行必要的更正。 以下是一个可能的解决方案: ```python class SSDHead(nn.Module): def __init__(self, num_classes, in_channels, feat_channels=256, stacked_convs=2, **kwargs): super(SSDHead, self).__init__(**kwargs) self.num_classes = num_classes self.in_channels = in_channels self.feat_channels = feat_channels self.stacked_convs = stacked_convs self.loss_cls = nn.CrossEntropyLoss() # 定义loss_cls属性 self.loss_bbox = nn.L1Loss(reduction='none') self.conv1x1 = nn.ModuleList() self.conv3x3 = nn.ModuleList() for i in range(self.stacked_convs): self.conv1x1.append(nn.Conv2d(self.in_channels, self.feat_channels, kernel_size=1)) self.conv3x3.append(nn.Conv2d(self.feat_channels, self.feat_channels, kernel_size=3, padding=1)) self.cls_convs = nn.ModuleList() self.reg_convs = nn.ModuleList() for i in range(4): self.cls_convs.append(nn.Conv2d(self.feat_channels, self.feat_channels, kernel_size=3, padding=1)) self.reg_convs.append(nn.Conv2d(self.feat_channels, self.feat_channels, kernel_size=3, padding=1)) self.cls_out = nn.Conv2d(self.feat_channels, self.num_classes, kernel_size=3, padding=1) self.reg_out = nn.Conv2d(self.feat_channels, 4, kernel_size=3, padding=1) def forward(self, x): cls_scores = [] bbox_preds = [] for feat in x: cls_feat = feat reg_feat = feat for i in range(self.stacked_convs): cls_feat = F.relu(self.conv1x1[i](cls_feat)) cls_feat = F.relu(self.conv3x3[i](cls_feat)) reg_feat = F.relu(self.conv1x1[i](reg_feat)) reg_feat = F.relu(self.conv3x3[i](reg_feat)) cls_feat = cls_feat + feat reg_feat = reg_feat + feat cls_feat = self.cls_convs[0](cls_feat) reg_feat = self.reg_convs[0](reg_feat) for i in range(1, 4): cls_feat = F.relu(cls_feat) reg_feat = F.relu(reg_feat) cls_feat = self.cls_convs[i](cls_feat) reg_feat = self.reg_convs[i](reg_feat) cls_score = self.cls_out(cls_feat) bbox_pred = self.reg_out(reg_feat) cls_scores.append(cls_score) bbox_preds.append(bbox_pred) return cls_scores, bbox_preds def loss(self, cls_scores, bbox_preds, gt_bboxes, gt_labels, img_metas): losses = dict() batch_size = cls_scores[0].size(0) featmap_sizes = [featmap.size()[-2:] for featmap in cls_scores] device = cls_scores[0].device gt_bboxes = [gt_bbox.to(device) for gt_bbox in gt_bboxes] gt_labels = [gt_label.to(device) for gt_label in gt_labels] anchor_list, valid_flag_list = self.get_anchors(featmap_sizes, img_metas, device=device) cls_reg_targets = self.anchor_target(anchor_list, valid_flag_list, gt_bboxes, img_metas, gt_labels) if cls_reg_targets is None: return None (labels_list, label_weights_list, bbox_targets_list, bbox_weights_list, num_total_pos, num_total_neg) = cls_reg_targets num_total_samples = reduce_mean([labels.size(0) for labels in labels_list]) losses['loss_cls'] = self.loss_cls(cls_scores, labels_list, label_weights_list) # 调用loss_cls属性 losses['loss_bbox'] = self.loss_bbox(bbox_preds, bbox_targets_list, bbox_weights_list, avg_factor=num_total_samples) return losses ```

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

有情怀的机械男

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

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

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

打赏作者

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

抵扣说明:

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

余额充值