- 19 May, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Adding an MLP block. * Adding documentation * Update typos. * Fix inplace for Dropout. * Apply recommendations from code review. * Making changes on pre-trained models. * Fix linter
-
- 18 May, 2022 1 commit
-
-
Aditya Oke authored
* try re arranging docs * rewrite docs a bit * Apply suggestions from code review Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> 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>
-
- 27 Feb, 2022 1 commit
-
-
Hu Ye authored
* Create dropblock.py * add dropblock2d * fix pylint * refactor dropblock * add dropblock * Rename dropblock.py to drop_block.py * fix pylint * add dropblock * add dropblock3d * add drop_block3d * add dropblock * Update drop_block.py * Update torchvision/ops/drop_block.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update torchvision/ops/drop_block.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update torchvision/ops/drop_block.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update torchvision/ops/drop_block.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update drop_block.py * Update drop_block.py * import torch.fx * fix lint * fix lint * Update drop_block.py * improve dropblock * add dropblock * refactor dropblock * fix doc * remove the limitation of block_size * Update torchvision/ops/drop_block.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * fix lint * fix lint * add dropblock * Fix linter * add dropblock random check * reduce test time * Update test_ops.py * speed the dropblock test * fix lint Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 25 Feb, 2022 1 commit
-
-
Aditya Oke authored
* Add ops.conv3d * Refactor for conv2d and 3d * Refactor * Fix bug * Addres review * Fix bug * nit fix * Fix flake * Final fix * remove documentation * fix linter * Update all the implementations to use new Conv * Small doc fix Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Joao Gomes <jdsgomes@fb.com>
-
- 24 Feb, 2022 1 commit
-
-
Joao Gomes authored
-
- 02 Feb, 2022 1 commit
-
-
Aditya Oke authored
* Improve giou loss * Fix red Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 02 Nov, 2021 1 commit
-
-
Joao Gomes authored
* Expose misc ops at package level * Adding documentation to the ops exposed
-
- 01 Nov, 2021 1 commit
-
-
Nicolas Hug authored
-
- 29 Sep, 2021 1 commit
-
-
Aditya Oke authored
* start writing example * Update example * Add PenFudan Files * Update example * Remove unused files * Update file and adopt changes * Create links, fix float Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
- 21 Sep, 2021 1 commit
-
-
Allen Goodman authored
* ops.masks_to_bounding_boxes * test fixtures * unit test * ignore lint e201 and e202 for in-lined matrix * ignore e121 and e241 linting rules for in-lined matrix * draft gallery example text * removed type annotations from pytest fixtures * inlined fixture * renamed masks_to_bounding_boxes to masks_to_boxes * reformat inline array * import cleanup * moved masks_to_boxes into boxes module * docstring cleanup * updated docstring * fix formatting issue * gallery example * use torch * use torch * use torch * use torch * updated docs and test * cleanup * updated import * use torch * Update gallery/plot_repurposing_annotations.py Co-authored-by:
Aditya Oke <47158509+oke-aditya@users.noreply.github.com> * Update gallery/plot_repurposing_annotations.py Co-authored-by:
Aditya Oke <47158509+oke-aditya@users.noreply.github.com> * Update gallery/plot_repurposing_annotations.py Co-authored-by:
Aditya Oke <47158509+oke-aditya@users.noreply.github.com> * Autodoc * use torch instead of numpy in tests * fix build_docs failure * Closing quotes. Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Aditya Oke <47158509+oke-aditya@users.noreply.github.com>
-
- 20 Aug, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Adding operator. * Adding tests * switching order of `p` and `mode`. * Remove seed setting. * Replace stats import with pytest.importorskip. * Fix doc * Apply suggestions from code review Co-authored-by:
Francisco Massa <fvsmassa@gmail.com> * Fixing indentation. * Adding operator in the documentation. * Fixing lint Co-authored-by:
Francisco Massa <fvsmassa@gmail.com>
-
- 10 Nov, 2020 1 commit
-
-
Aditya Oke authored
-
- 01 Oct, 2020 1 commit
-
-
Aditya Oke authored
* adds boxes conversion * adds documentation * adds xywh tests * fixes small typo * adds tests * Remove sphinx theme * corrects assertions * cleans code as per suggestion Signed-off-by:
Aditya Oke <okeaditya315@gmail.com> * reverts assertion * fixes to assertEqual * fixes inplace operations * Adds docstrings * added documentation * changes tests * moves code to box_convert * adds more tests * Apply suggestions from code review Let's leave those changes to a separate PR * fixes documentation Co-authored-by:
Francisco Massa <fvsmassa@gmail.com>
-
- 24 Sep, 2020 1 commit
-
-
Aditya Oke authored
* tries adding genaralized_iou * fixes linting * Adds docs for giou, iou and box area * fixes lint * removes docs to fixup in other PR * linter fix * Cleans comments * Adds tests for box area, iou and giou * typo fix for testCase * fixes typo * fixes box area test * fixes implementation * updates tests to tolerance
-
- 18 Sep, 2020 1 commit
-
-
Aditya Oke authored
* Adds docs and fixes improts for ops and io * reverts io and init for images
-
- 23 Jun, 2020 2 commits
-
-
Zhiqiang Wang authored
* Update docs * Update docs/source/ops.rst Co-authored-by:
Francisco Massa <fvsmassa@gmail.com> Co-authored-by:
Francisco Massa <fvsmassa@gmail.com>
-
Zhiqiang Wang authored
* Add Deformable Convolution in Doc * Add PSRoIPool and PSRoIAlign in Doc
-
- 02 Aug, 2019 1 commit
-
-
Francisco Massa authored
* Expose docs for io and ops package Had do modify the docstrings to use Napoleon NumPy style, because Napoleon Google Style doesn't support multiple return arguments * Add video section
-