- 05 Oct, 2022 1 commit
-
-
Aditya Gandhamal authored
* Add ValueError * Add tests for ValueError * Add tests for ValueError * Add ValueError * Change to if/else * Ammend iou_fn tests * Move code excerpt * Format tests Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 18 May, 2022 1 commit
-
-
Aditya Oke authored
* Cleanup ops * Address nits Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 09 May, 2022 1 commit
-
-
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:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 28 Apr, 2022 1 commit
-
-
Abhijit Deo authored
* added ciou loss * "formatting with flake8 and ufmt" * formatting with ufmt and flake8 * minor changes * changes as per the suggestions * added reference in torchvision/ops/__init__.py * sample test * tests formatted * added description * formatting * edited tests * changes in tests * added tests for multiple boxes * minor edits * minor edit * doc added * minor edits * Update test_ops.py * formatting test file * changes as per the suggestions * formatting and adding some more tests * bounding box added * removed unnecessary comment * added docstring * added type annotations * removed potential bug * Update torchvision/ops/boxes.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update torchvision/ops/boxes.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update test/test_ops.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Philip Meier <github.pmeier@posteo.de>
-