不能得到任何的bbox
sunlanchang opened this issue · 5 comments
sunlanchang commented
在我训练完模型之后,尝试预测bbox,可是在预测bbox时候,也就是运行predict.py
文件加载训练好的模型best.pth
,不能得到任何的bbox,在查看代码后发现mask1 = contain > 0.1 # 大于阈值
这一行代码的mask1[:,:,0]
为0,如下图所示:
tensor([[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=torch.uint8)
请问我的问题大概出在哪里呢?
han1366 commented
我跟你一样,你解决了吗
sunlanchang commented
我跟你一样,你解决了吗
没有解决。
HusterRC commented
@sunlanchang 好好研究以下作者计算loss的代码,可能发现问题
yunshangyue71 commented
@sunlanchang 好好研究以下作者计算loss的代码,可能发现问题
作者loss, confidence 是不是和paper 不一致啊
Vandaci commented
作者的loss好像是把非”responsible“的bndbox误差也算进去了