GitLab

* add xfailed smoke tests for dispatchers

* also support old FunctionalInfo's

* try reduce memory consumption for CI

* fix sample inputs generation
Co-authored-by: vfdev <vfdev.5@gmail.com>

* Updated fill arg typehint for affine, perspective and elastic ops

* Updated pad op on prototype side

* Code updates

* Few other minor updates

* WIP

* WIP

* Updates

* Update _image.py

* Fixed tests

* Updated fill arg typehint for affine, perspective and elastic ops

* Updated pad op on prototype side

* Code updates

* Few other minor updates

Not sure what have changed between CUDA-11.3 and CUDA-11.6, but bumping tolerance keeps tests green

* fix convert_format_bounding_box, affine_bounding_box, and resize_bounding_box

* fix perspective and elastic

* fix tests

* fix PIL and tensor mask comparison

* introduce kernel_name field

* add dtype consistency test

* port some tests from old framework

* add kernel infos for conversion kernels

* cleanup

* use nearest and bicubic for resize image sample inputs

* make parametrization id more obvious

* use named sentinel instead of None for random image size

* crestereo draft implementation

* minor model fixes. positional embedding changes.

* aligned base configuration with paper

* Adressing comments

* Broke down Adaptive Correlation Layer. Adressed some other commets.

* adressed some nits

* changed search size, added output channels to model attrs

* changed weights naming

* changed from iterations to num_iters

* removed _make_coords, adressed comments

* fixed jit test

* config nit

* Changed device arg to str
Co-authored-by: Joao Gomes <jdsgomes@fb.com>
Co-authored-by: YosuaMichael <yosuamichaelm@gmail.com>

* [proto] Added consistency tests for detection transforms

* Updated tests according to the review

* More updates
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

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

* reduce verbosity of prototype tests and split them into multiple steps

* rename builtin datasets module to fit common scheme

* Rename `convert_bounding_box_format` => `convert_format_bounding_box`

* Add missed replacement.

* Fix bug on `get_image_size()` and move it to deprecated. Introduce generic named spatial/channel equivalents.

* Update tests and fix mypy issues.

* Remove the use of get_image_size from ElasticTransform.

* Fix linter

* Apply suggestions from code review.

* Update torchvision/prototype/transforms/functional/_deprecated.py
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Further changes from code review.

* Fix linter
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* rename features.SegmentationMask -> features.Mask

* rename kernels *_segmentation_mask -> *_mask and cleanup input name

* cleanup

* rename module _segmentation_mask.py -> _mask.py

* fix test

* add generator functions for segmentation masks

* update functional tests and fix kernels

* fix transforms tests

* port RandomResize from segmentation references

* mypy

* remove optional max_size

* add interpolation and antialias

* [SKIP CI] only CircleCI

* fix bug

* [SKIP CI] only CircleCI

* [SKIP CI] add test

* Add `handle_legacy_interface()` to all new models.

* Fix imports

* Addressing review comments.

* Fix linter

* Addressing further comments.

* Added Middlebury2014   dataset

* adressed nits

* mypy fix

Co-authored-by: Joao Gomes <jdsgomes@fb.com>

* avoid double padding parsing

* remove cloning in degenerate case

* fix affine and rotate for degenerate inputs

* fix rotate for degenerate inputs if expand=True

* fix padding for degenerate segmentation masks

* extend test data degeneration to degenerate inputs

* add even more degenerate shapes

* simplify kernel

* [SKIP CI] only GHA

* add more degenerate segmentation masks

* fix segmentation mask generation

* xfail some tests

* Revert "simplify kernel"

This reverts commit 18c5e4fc59de86fdfd304bf0c0a988b410b81fd9.

* fix resize for degenerate inputs

* [SKIP CI] CircleCI

* fix RandomIoUCrop test

* [SKIP CI] CircleCI

* cleanup

* [SKIP CI] CircleCI

* add perf TODO comments

* [SKIP CI] CircleCI

* [proto] Fixed bug in ScaleJitter with params

* Updated tests

* Skip big models per platform/device

* Specifying skips on Windows only.

* Simplify and clean up code.

* update roipool to make in torch fx traceable

* update ps_roi_align, ps_roi_pool and roi_align to make them torch fx traceable

* add unittest

* Remove `.to(output_fx)` from test
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* check signatures of all available transformations

* fix signatures of RandAugment and TrivialAugmentWide

* move AA consistency tests to correct module

* add test for signature consistency

* fix CenterCrop and Lambda

* add removed params

* cleanup

* add consistency tests for prototype container transforms

* fix RandomApply

* clean up bboxes

* Correcting bug on RandAugment.

* Fix bug on erase()

* Moving `_parse_pad_padding()` to functional geometry

* Fixing bug on deprecated Grayscale transforms

* Porting old bugfixes from main branch to _apply_image_transform

* Dropping mandatory keyword arguments to maintain BC.

* Adding antialias option where possible.

* Specifying types in `_transform()` where possible.

* Add todo.

* Fixing tests.

* Adding padding_mode in pad_bbox
Co-authored-by: vfdev <vfdev.5@gmail.com>

* [proto] Fixed RandAug implementation

* Fixed randomness in tests for trivial aug

* Fixed all AA tests

* add checks for pure random apply transformations

* add tests for some non-deterministic transforms used in classification reference

* add consistency checks for most transforms

* fix RandomAffine, RandomPerspective, and GaussianBlur

* fix tests

* add test for coverage

* add tests for remaining deterministic transforms

* fix Lambda

* improve test framework and add more consistency configs

* improve error messages

* increase coverage for Resize

* improve comparison

* add consistency checks for prototype and legacy transforms

* fix Resize