GitLab

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

Co-authored-by: Nicolas Hug <contact@nicolas-hug.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

- call torch.log once can save few ms
- all masks should be selected with is_within_crop_area in RandomIoUCrop
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

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

* Remove alias for `pil_to_tensor()`

* Remove alias for `PILToTensor`

* Leave `to_tensor()` as-is.

* Fix linter

* 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>

Fix cuda 11.6 and 11.7

Check nvjpeg from CONDA Path

More cuda11.6

Trying to fix the build

Trying to fix the build

Removing nvjpeg logic

Add NvToolsExt dll path

Fix lint

* Update raft_stereo to sync forward param with CREStereo and have output_channel

* Add flow_init param to docstring

* Use output_channels instead of output_channel

* Replace depth with disparity since what we predict actually disparity instead of actual depth

* rename features.SegmentationMask -> features.Mask

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

* cleanup

* rename module _segmentation_mask.py -> _mask.py

* fix test

* fix resize for segmentation masks without batch dim

* micro improvement of stable crop

* Revert "micro improvement of stable crop"

This reverts commit e981e36f92dabc8f2ee05bec53a628a476a2b236.

Not sure, why it wasn't the case before

* add generator functions for segmentation masks

* update functional tests and fix kernels

* fix transforms tests

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

* 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