search

keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras
Other
1.01k stars 330 forks source link

IoU Loss #113

Closed LukeWood closed 2 years ago

LukeWood commented 2 years ago

IoU loss is a simple, easy to understand bounding box loss function

Myself or Scott will probably write this to set the example for loss functions

bhack commented 2 years ago

https://github.com/tensorflow/models/blob/master/official/vision/beta/projects/yolo/losses/yolo_loss.py

quantumalaviya commented 2 years ago

@LukeWood I can work on this loss as well!

abhi-glitchhg commented 2 years ago

Hey there! Any plans to add Generalised IOU(GIOU) - https://arxiv.org/abs/1902.09630 ; Complete IOU loss (CIOU) and Distance IOU loss (DIOU) - https://arxiv.org/abs/1911.08287

Thanks cc : @LukeWood

quantumalaviya commented 2 years ago

Hey @abhi-glitchhg, I am planning on adding these losses soon.

LukeWood commented 2 years ago

@quantumalaviya added this