训练完之后测试模型，模型没有角度回归呢

[zqs1996]

[yzqxy]

看看数据标注有没有问题，增加训练轮次，调整下训练的batch

[zspzhangshoupeng]

怎么样，有所改观没？角度

[Tristesse-stk]

同角度没有回归

[cl886699]

同角度没有回归

看代码loss计算里的角度loss是被注释掉的

[Tristesse-stk]

同角度没有回归

看代码loss计算里的角度loss是被注释掉的

我后来看到了，把注释部分还原回去训练，角度还是水平的方向

[yzqxy]

角度loss用的是probiou loss，注释的是kld loss ,效果不如probiou loss。如果是之前下再的代码，loss里面这个地方改一下box_loss +=(probloss weight).sum() / (target_scores_sumbatch_size)。然后再训试试

[Tristesse-stk]

角度loss用的是probiou loss，注释的是kld loss ,效果不如probiou loss。如果是之前下再的代码，loss里面这个地方改一下box_loss +=(probloss weight).sum() / (target_scores_sumbatch_size)。然后再训试试

我两个都用过了，训练完之后模型的角度都是水平的。