GitLab

* [proto] Clean-up Label.to_categories

* Fixed flake8

* Use --no-implicit-optional for type checking

This is needed for PEP 484 compliant type checking (as of 2018).
mypy will change its defaults soon.

See https://github.com/python/mypy/issues/9091



* forcing CI rerun
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* disable download test for Places365

* add xfail test for broken download servers

* add more comments

* mark SBU download as failing
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* [proto] Another attemp to rewrite RandomCrop

* Fixed implementation issue and updated tests

* shlex -> shutil

* [SKIP CI]

* [skip ci]
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Testing conda metapackage

* Testing

* Testing pytorch-cuda

* Testing

* Testing

* Testing

* Testing

* Testing

* Using new metapackage

* Testing

* Simplifing logic

* Further simplifying

* [proto] Compose keeps BC

* Compose -> Compose(Transform)

* Fix generated CMake file

* Make optional dependencies consistent
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Fix typing jit issue on RoIPool and RoIAlign

* Fix nit.

* Address code review comments.

* [proto] Use F.pad in RandomCrop

* Update torchvision/prototype/transforms/_geometry.py

* init submit

* fix typo

* support ufmt and mypy

* fix 2 unittest errors

* fix ufmt issue

* Apply suggestions from code review
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* unify codes

* fix meshgrid indexing

* fix a bug

* fix type check

* add type_annotation

* add slow model

* fix device issue

* fix ufmt issue

* add expect pickle file

* fix jit script issue

* fix type check

* keep consistent argument order

* add support for pretrained_window_size

* avoid code duplication

* a better code reuse

* update window_size argument

* make permute and flatten operations modular

* add PatchMergingV2

* modify expect.pkl

* use None as default argument value

* fix type check

* fix indent

* fix window_size (temporarily)

* remove "v2_" related prefix and add v2 builder

* remove v2 builder

* keep default value consistent with official repo

* deprecate dropout

* deprecate pretrained_window_size

* fix dynamic padding edge case

* remove unused imports

* remove doc modification

* Revert "deprecate dropout"

This reverts commit 8a13f932815ae25655c07430d52929f86b1ca479.

* Revert "fix dynamic padding edge case"

This reverts commit 1c7579cb1bd7bf2f0f94907f39bee6ed707a97a8.

* remove unused kwargs

* add downsample docs

* revert block default value

* revert argument order change

* explicitly specify start_dim

* add small and base variants

* add expect files and slow_models

* Add model weights and documentation for swin v2

* fix lint

* fix end of files line
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Joao Gomes <jdsgomes@fb.com>

* Extending to support MViTv2

* Fix docs, mypy and linter

* Refactor the relative positional code.

* Code refactoring.

* Rename vars.

* Update docs.

* Replace assert with exception.

* Updat docs.

* Minor refactoring.

* Remove the square input limitation.

* Moving methods around.

* Modify the shortcut in the attention layer.

* Add ported weights.

* Introduce a `residual_cls` config on the attention layer.

* Make the patch_embed kernel/padding/stride configurable.

* Apply changes from code-review.

* Remove stale todo.

* [proto] Fixed issue with `F.pad` from RandomZoomOut

* Fixed failing tests

* Fixed wrong type hint

* Fixed fill=None in pad_image_pil

* Try to support fill=None in functional

* Code formatting

* [proto] Fix for handling Numpy arrays by Transform

* transformed_types -> _transformed_types

* Added Image.normalize and fixed Normalize._transform

* Updated code for normalize, removed Image.normalize

* Expose on Hub the public methods of the registration API

* Limit methods and update docs.

* Update references to use the new Model Registration API

* Expose on Hub the public methods of the registration API

* Limit methods and update docs.

* Make test precision stricter for Classification

* Update classification threshold.

* Update quantized classification threshold.

Update `RandomPhotometricDistort` `__init__` argument to correct types.

* added SceneFLow variant datasets

* Changed split name to variant name

* removed trailing commented code line

Summary: provider callback uses video sampler to copy/scale input frames using sws_scale function, added option to use av_image_copy to copy image planes in case if sws_getContext fails (required to support AV_PIX_FMT_PAL8 pixel format)

Reviewed By: ovoietsa

Differential Revision: D36466749

fbshipit-source-id: 1674f98a362db835d147800c24bdda6f82d2f8ff
Co-authored-by: Sergiy Bilobrov <sergiy@fb.com>

Summary:
Resolves https://www.internalfb.com/tasks/?t=128004042

Caused by TorchVision's PR at https://github.com/pytorch/vision/pull/6322 which was in response to a change on PyTorch Core at https://github.com/pytorch/pytorch/pull/82032

Reviewed By: fmassa

Differential Revision: D38383266

fbshipit-source-id: 3f0ebd04a610031c4720123d1869a851f76455cd

* Broken down PR(#6269). Added an additional dataset

* Removed some types. Store None instead of "". Merged test util functions.

* minor mypy fixes. minor doc fixes

* reformated docstring

* Added additional line-skips

* Fix bug in calculating cIoU for unequal sizes

* Remove comment

* what the epsilon?

* Fixing DIoU

* Optimization by Francisco.

* Fix the expected values on CompleteBoxIoU

* Apply suggestions from code review
Co-authored-by: Abhijit Deo <72816663+abhi-glitchhg@users.noreply.github.com>

* Adding cartesian product test.

* remove static
Co-authored-by: Vasilis Vryniotis <vvryniotis@fb.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Abhijit Deo <72816663+abhi-glitchhg@users.noreply.github.com>

* cleaning up box decoding

* minor nits

* cleanup for box encoding also addded.

* Model registration mechanism.

* Add overwrite options to the dataset prototype registration mechanism.

* Adding example models.

* Fix module filtering

* Fix linter

* Fix docs

* Make name optional if same as model builder

* Apply updates from code-review.

* fix minor bug

* Adding getter for model weight enum

* Support both strings and callables on get_model_weight.

* linter fixes

* Fixing mypy.

* Renaming `get_model_weight` to `get_model_weights`

* Registering all classification models.

* Registering all video models.

* Registering all detection models.

* Registering all optical flow models.

* Fixing mypy.

* Registering all segmentation models.

* Registering all quantization models.

* Fixing linter

* Registering all prototype depth perception models.

* Adding tests and updating existing tests.

* Fix linters

* Fix tests.

* Add beta annotation on docs.

* Fix tests.

* Apply changes from code-review.

* Adding documentation.

* Fix docs.

[FBcode->GH]
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/81946



Reviewed By: linbinyu

Differential Revision: D38056789

fbshipit-source-id: 6a2e2d3308c9eec5b2230e153c5e2863b3b5bf7b
Co-authored-by: Sim Sun <simsun@fb.com>

* [proto] Added few transforms tests, part 1 (#6262)

* Added supported/unsupported data checks in the tests for cutmix/mixup

* Added RandomRotation, RandomAffine transforms tests

* Added tests for RandomZoomOut, Pad

* Update test_prototype_transforms.py

* Added RandomCrop transform and tests (#6271)

* [proto] Added GaussianBlur transform and tests (#6273)

* Added GaussianBlur transform and tests

* Fixing code format

* Copied correctness test

* [proto] Added random color transforms and tests (#6275)

* Added random color transforms and tests

* Disable smoke test for RandomSolarize, RandomAdjustSharpness

* Added RandomPerspective and tests (#6284)

- replaced real image creation by mocks for other tests

* Added more functional tests (#6285)

* [proto] Added elastic transform and tests (#6295)

* WIP [proto] Added functional elastic transform with tests

* Added more functional tests

* WIP on elastic op

* Added elastic transform and tests

* Added tests

* Added tests for ElasticTransform

* Try to format code as in https://github.com/pytorch/vision/pull/5106



* Fixed bug in affine get_params test

* Implemented RandomErase on PIL input as fallback to tensors (#6309)

Added tests

* Added image_size computation for BoundingBox.rotate if expand (#6319)

* Added image_size computation for BoundingBox.rotate if expand

* Added tests

* Added erase_image_pil and eager/jit erase_image_tensor test (#6320)

* Updates according to the review
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Refactor tests

* Remove tol, fix comments

* Add tolerance only where necessary

* Add tolerance only where necessary

* Add tolerance only where necessary

* Refactor to adapt suggestions

* Refactor and add nits

* Refactor box area

* Refactor to one file

* Adapt almost all except area

* final update

* Tighten for jit

* Refactor slightly

* Fix tests
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Indicate strings can be used to specify

* quotes

Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* [proto] Added missing decorator for center_crop_segmentation_mask tests

* Added center_crop_image_tensor