keras-team / keras-cv

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

Adding DIoU loss and CIoU Loss to keras-cv #1736

Open Anirban0011 opened 1 year ago

Anirban0011 commented 1 year ago

Should this two losses be added to keras_cv.losses ?

  1. DIoU (Distance over IoU) Loss is independent to the size of the object being detected. It is seen to converge mush faster than GIoU even with opposite orientation of boxes(vertical and horizontal). It is also deployed as a criterion in NMS layer of which we already have an implementation in the repo.

  2. CIoU (Complete IoU) loss needs fewer iteration than both GIoU and DIoU loss to converge. It has been deployed in YOLO v3, Yolo v4, SSD, and Faster RCNN.

Anirban0011 commented 1 year ago

A paper which highlights the limitations of existing BBR loss functions : Focal and Efficient IOU Loss for Accurate Bounding Box Regression

jaygala223 commented 1 year ago

Hello @ianstenbit, can you please take a look at this issue? I would like to work on this