pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
16.13k stars 6.94k forks source link

Clean up Ops Box and Losses implementations #5976

Closed datumbox closed 2 years ago

datumbox commented 2 years ago

🚀 The feature

Currently losses such as cIoU, dIoU and gIoU reimplement functions that exist already on boxes ops. In addition they contain a lot of duplicate code that can be shared. Finally the testing strategy can be improved.

This ticket is about:

cc @oke-aditya

oke-aditya commented 2 years ago

I'm on this :smile: (I think as per Phillip's comment I can't self assign, but feel free to assign me)