• Yassine Alouini's avatar
    Distance IoU (#5786) · 1ae38297
    Yassine Alouini authored
    
    
    * [FEAT] Add distance IoU and distance IoU loss + some tests (WIP for tests).
    
    * [FIX] Remove URL from docstring + remove assert since it causes a big performance drop.
    
    * [FIX] eps isn't None.
    
    * [TEST] Update existing box dIoU test + add dIoU loss tests (inspired from cIoU ones).
    
    * [ENH] Some pre-commit fixes + remove print + mypy.
    
    * [ENH] Pass the device in the assertion for the dIoU loss test.
    
    * [FIX] Remove type hints from the dIoU box test.
    
    * [ENH] Refactor box and loss for dIoU functions + fix half tests.
    
    * [FIX] Precommits fix.
    
    * [ENH] Some improvement for the distance IoU tests thanks to code review.
    
    * [ENH] Upcast in distance boxes computation to avoid overflow.
    
    * [ENH] Revert the refactor of distance IoU loss back since it introduced a bug and can be slow.
    
    * Precommit fix.
    
    * [FIX] Few changes introduced by merge conflict.
    
    * Add code reference
    
    * Fix test
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    1ae38297
ops.rst 1.03 KB