ssds目标检测项目中遇到的一些bug

1.问题
在ssds检测项目中,其中求损失的multibox_loss.py中遇到一个bug,错误为:

IndexError: The shape of the mask [32, 2990] at index 0 does not match the shape of the indexed tensor [95680, 1] at index 0

报错代码如下:

loss_c[pos] = 0

其中,pos和loss_c的尺寸维度分别是:

loss_c.size torch.Size([95680, 1])
pos.size torch.Size([32, 2990])

然后,我写了一小段代码,来看一下loss_c[pos]操作是怎么操作的,其中的原理以及作用是什么。

测试代码:

loss = torch.rand(10,1)
print('loss',loss)
pos = pos.byte()
print('pos.size',pos.size(),'pos.type',type(pos),'\n','pos',pos)
print(loss[pos])

结果如下:
在这里插入图片描述
可以看出,如果直接用上述的pos和loss_c的维度是会报上述的错误,经过我的测试,loss_c[pos]操作是把pos为1的元素对应的位置上

  • 14
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 8
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值