- 09 Dec, 2021 6 commits
-
-
Kai Zhang authored
* revamp log api usage method
-
Nicolas Hug authored
-
Nicolas Hug authored
-
Nicolas Hug authored
-
Nicolas Hug authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Nicolas Hug authored
-
- 08 Dec, 2021 8 commits
-
-
Nicolas Hug authored
-
Nicolas Hug authored
-
Philip Meier authored
-
Nicolas Hug authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Nicolas Hug authored
-
Vasilis Vryniotis authored
* Adding test that verifies meta schema. * Adding missing interpolation to keypoint model. * Change assertion. * Adding map metric in meta schema for detection and renaming metrics for kp and mask rcnn.
-
Philip Meier authored
* add download functionality to prototype datasets * fix annotation * fix test * remove iopath * add comments
-
Philip Meier authored
* Revert "[FBcode->GH] remove unused requests functionality (#5014)" This reverts commit 33123bee. * Revert "replace requests with urllib (#4973)" This reverts commit 8d25de7b. * add requests as hard dependency * install library stubs in CI * fix syntax * add requests to conda dependencies * fix mypy CI
-
- 07 Dec, 2021 2 commits
-
-
Vasilis Vryniotis authored
* Adding logging calls for raft and vit * Linter fix
-
Philip Meier authored
* simplify model builders * cleanup * refactor kwonly to pos or kw handling * put weight verification back * revert num categories checks * fix default weights * cleanup * remove manual parameter map * refactor decorator interface * address review comments * cleanup * refactor callable default * fix type annotation * process ungrouped models * cleanup * mroe cleanup * use decorator for detection models * add decorator for quantization models * add decorator for segmentation models * add decorator for video models * remove old helpers * fix resnet50 * Adding verification back on InceptionV3 * Add kwargs in DeeplabeV3 * Add kwargs on FCN * Fix typing on Deeplab * Fix typing on FCN Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 06 Dec, 2021 1 commit
-
-
Nicolas Hug authored
-
- 05 Dec, 2021 1 commit
-
-
Prabhat Roy authored
-
- 03 Dec, 2021 1 commit
-
-
Philip Meier authored
* exclude windows from mmap * add comment for windows * appease mypy
-
- 02 Dec, 2021 2 commits
-
-
Philip Meier authored
-
Vasilis Vryniotis authored
-
- 01 Dec, 2021 3 commits
-
-
Philip Meier authored
-
Yuxin Wu authored
According to the benchmark link https://github.com/pytorch/vision/issues/1311#issuecomment-781329339 , for GPU the threshold should be higher (which is why detectron2 used 40k). This PR changes the threshold to be device dependent. Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> Co-authored-by:
Francisco Massa <fvsmassa@gmail.com>
-
Nicolas Hug authored
* Add bias parameter to ConvNormActivation * Update torchvision/ops/misc.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 30 Nov, 2021 4 commits
-
-
Philip Meier authored
Co-authored-by:Prabhat Roy <prabhatroy@fb.com>
-
Joao Gomes authored
making torchvision ops leaf nodes by default
-
Philip Meier authored
* improve COCO prototype * test 2017 annotations * add option to include captions * fix categories and add tests * cleanup * add correct image size to bounding boxes * fix annotation collation * appease mypy * add benchmark * always use image as reference * another refactor * add support for segmentations * add support for segmentations * fix CI dependencies
-
Vasilis Vryniotis authored
* Change the `default` weights mechanism to sue Enum aliases. * Change `get_weights` to work with full Enum names and make it public. * Applying improvements from code review.
-
- 29 Nov, 2021 4 commits
-
-
Vasilis Vryniotis authored
* Fix bug on autocontrast when `min==max` * Adding PIL vs TorchVision test for min==max
-
Vasilis Vryniotis authored
* Rename classes Weights => WeightsEnum and WeightEntry => Weights. * Make enum values follow the naming convention `_V1`, `_V2` etc * Cleanup the Enum class naming conventions. * Add a test to check naming conventions.
-
Nicolas Hug authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
puhuk authored
* Update functional_tensor.py To resolve issue #4818 Add assert function and logic after checking bound of image * Update functional_tensor.py * Update test * Update test_functional_tensor.py * Update test_functional_tensor.py * Update test_functional_tensor.py * Update test_functional_tensor.py * Update test_functional_tensor.py * Update test_functional_tensor.py * Update test_functional_tensor.py * Update test_functional_tensor.py * Fix linter Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 27 Nov, 2021 1 commit
-
-
Yiwen Song authored
* [vit] Adding ViT to torchvision/models * adding pre-logits layer + resolving comments * Fix the model attribute bug * Change version to arch * fix failing unittests * remove useless prints * reduce input size to fix unittests * Increase windows-cpu executor to 2xlarge * Use `batch_first=True` and remove classifier * Change resource_class back to xlarge * Remove vit_h_14 * Remove vit_h_14 from __all__ * Move vision_transformer.py into prototype * Fix formatting issue * remove arch in builder * Fix type err in model builder * address comments and trigger unittests * remove the prototype import in torchvision.models * Adding vit back to models to trigger CircleCI test * fix test_jit_forward_backward * Move all to prototype. * Adopt new helper methods and fix prototype tests. * Remove unused import. Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Vasilis Vryniotis <vvryniotis@fb.com>
-
- 26 Nov, 2021 1 commit
-
-
Philip Meier authored
-
- 25 Nov, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Updated the link for densenet recipe. * Set default value of `num_classes` and `num_keypoints` to `None` * Provide helper methods for parameter checks to reduce duplicate code. * Throw errors on silent config overwrites from weight meta-data and legacy builders. * Changing order of arguments + fixing mypy. * Make the builders fully BC. * Add "default" weights support that returns always the best weights.
-
- 23 Nov, 2021 1 commit
-
-
Philip Meier authored
-
- 22 Nov, 2021 4 commits
-
-
F-G Fernandez authored
* style: Fixed last missing typing annotation * style: Fixed typing * style: Fixed remaining typing annotations * style: Fixed typing * style: Fixed typing * refactor: Removed unused import * Update torchvision/datasets/video_utils.py Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
Prabhat Roy <prabhatroy@fb.com>
-
Philip Meier authored
* don't close file handles in protoype dataset tests * remove file closing from decoder Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Philip Meier authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Joao Gomes authored
* change to stable sort in nms implementations
-