yzqxy / Yolov8_obb_Prune_Track

GNU General Public License v3.0
181 stars 14 forks source link

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

Open zqs1996 opened 1 year ago

zqs1996 commented 1 year ago

image

yzqxy commented 1 year ago

image

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

zspzhangshoupeng commented 1 year ago

怎么样,有所改观没?角度

Tristesse-stk commented 1 year ago

同角度没有回归

cl886699 commented 1 year ago

同角度没有回归

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

Tristesse-stk commented 1 year ago

同角度没有回归

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

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

yzqxy commented 1 year ago

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

Tristesse-stk commented 1 year ago

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

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