RetinaNet_OBB+PolyGiouLoss在HRSC上无法训练

jbwang1997 / OBBDetection

OBBDetection is an oriented object detection library, which is based on MMdetection.

Apache License 2.0

522 stars 112 forks source link

Closed Xinbo-01 closed 2 years ago

Xinbo-01 commented 2 years ago

在HRSC数据集上使用retinanet_obb配合polygiouloss训练会直接nan，即使将学习率调为0也会nan掉

Xinbo-01 commented 2 years ago

jbwang1997 commented 2 years ago

我看一下，PolyIoULoss会有这个问题吗

Xinbo-01 commented 2 years ago

PolyIoULoss虽然可以跑起来，但是无法收敛

Xinbo-01 commented 2 years ago

还有一点疑惑是与iou loss.py代码相比，polyiouloss中并未将box decoder解码

jbwang1997 commented 2 years ago

解码过程时在dense_head里面进行解码的，需要设置reg_decoded_bbox=True. 这个不知道是否有设置？