search

CoinCheung / pytorch-loss

label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful
MIT License
2.17k stars 374 forks source link

Regarding Generalized IOU Loss #7

Closed samyak0210 closed 4 years ago

samyak0210 commented 4 years ago

Can G-IOU be used as a loss function? Actually I am training a bounding box prediction model and using GIOU as a loss function but after some few iterations of the first epoch of training, I am getting constant values.

CoinCheung commented 4 years ago

Yes, in the object detection field, and especially in these days, there are many paper exploring using iou-based loss for bounding box regression. The giou loss for object detection was proposed earlier, and you can find the paper here.

CoinCheung commented 4 years ago

I am closing this since you might have figured it out.