GitLab

* cleanup prototype transforms functional tests

* fix

* oust local functions

* port vertical_flip

* port rotate

* port crop

* port resized_crop

* port pad

* port perspective

* port elastic

* port center_crop

* port gaussian_blur

* port equalize

* port invert

* port posterize

* port solarize

* port autocontrast

* port adjust_sharpness

* port erase

* fix kernel infos

* renable crop references

* add tolerances to gaussian_blur kernel info

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

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

* Improve existing low kernel test.

* Add new midlevel jit-scriptability test (failing).

* Remove duplicate aliases from kernel tests.

* Fixing colour kernels.

* Fixing deprecated kernels.

* fix mypy

* Silence mypy instead of fixing to avoid performance penalty

* Fixing augment kernels.

* Fixing augment meta.

* Remove is_tracing calls.

* Add fake ImageType and DType

* Fixing type conversion kernels.

* Fixing misc kernels.

* partial fix geometry

* Remove mutable default from `_pad_with_vector_fill()` + all other unnecessary defaults.

* Fix geometry ops

* Fixing tests

* Removed xfail for jit tests on midlevel ops
Co-authored-by: vfdev-5 <vfdev.5@gmail.com>

* ADD: init_weights config for googlenet

* Fix: Inception and googlenet warnings

* Fix: warning in test_datasets.py

* Fix: Formatting error with ufmt

* Fix: Failing tests in quantized_classification_model

* Update test/test_models.py to make googlenet in 1 line
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Refactor: Change inception quantisation class initialization to use args/kwargs

* Resolve mypy issue

* Move *args before inception_blocks

* Move args keywords before other arguments
Co-authored-by: Ambuj Pawar <your_email@abc.example>
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* 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