GitLab

* 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

* [proto] Fix bug with Compose and PR 6504

* Added tests and fixed other bugs

* add aliases for hflip and vflip

* reduce imports from torchvision.transforms in torchvision.prototype.transforms

* add aliases for to_pil_image abd pil_to_tensor

* deprecate to_tensor

* add some FIXME cleanup comments

* address reviews

* add dimension getters

* undeprecate PILToTensor and ToPILImage

* address review

* fix test

* move simple_tensor to features module

* fix test

* check full sample in query_bounding_box

* support no bounding boxes in FixedSizeCrop

* fix test

* Move output to CPU before assert_expected

* Fixing `test_detection_model`

* check_device=False

* linter

[proto] Consistent supported/unsupported types handling in LinearTransformation, other perf comments (#6498)

* WIP

* [proto] unformal supported/unsupported types handling in LinearTransformation, other perf comments

* Type fixes and other minor stuff

* Fix pass-through and supported types of Normalize

* update error message on kernel

* Fix linter.

* Fix the tests.

* Update type.

* Update type.

* Remove unnecessary tests for bboxes and masks.

[proto] Use the proper `_transformed_types` in all Transforms and eliminate unnecessary dispatching (#6494)

* Update types in deprecated transforms.

* Update types in type conversion transforms.

* Fixing types in meta transforms.

* More changes on type conversion.

* Bug fix.

* Fix types

* Remove unnecessary conversions.

* Remove unnecessary import.

* Fixing tests

* Remove copy support from `to_image_tensor`

* restore test param

* Fix further tests

* [proto] Updated tests for CopyPaste on OneHotLabel

* Fixing test error

* Fixes unexpected behaviour with Transform._transformed_types and torch.Tensor

* Make code consistent to has_any, has_all implementation

* Fixed failing flake8 check

* add deprecated color conversion functionals

* allow grayscale tensor inputs in rgb_to_grayscale

* add cloning to tensor no-op

* improve todo comment

* [skip ci]

* use legacy kernels

use legacy kernels

* fix import

* remove duplicate code

* remove duplicate check

* Fix out-of-bounds read in decode_png

* Fix lint

* Use size_t instead of int to fix sign-compare check

* Add a unit-test to decode currupted png file

* Fix linter

* Update torchvision/csrc/io/image/cpu/decode_png.cpp

* Replace png_error with TORCH_CHECK
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Update error message

* Update error message
Co-authored-by: vfdev <vfdev.5@gmail.com>

* Fix linter
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: vfdev <vfdev.5@gmail.com>

* WIP

* [proto] Added SimpleCopyPaste transform

* Refactored and cleaned the implementation and added tests

* Fixing code

* Fixed code formatting issue

* Minor updates

* Fixed merge issue
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* add dispatcher for erase image kernels

* simplify RandomErasing

* replace query_image with query_image_dimensions

* use value sentinel and fix test for RandomErasing

* image_dimensions -> chw

* WIP

* Fixed dtype correction and tests

* Removed PIL Image support and output always Tensor

* [proto] Fixed to_image_tensor and to_image_pil

* Fixed failing test

* S3D initial commit

* add model builder code and docstrings

* change classifier submodule, populate weights enum

* fix change of block args from List[List[int]] to ints

* add VideoClassification to transforms

* edit weights url for testing, add s3d to models.video init

* norm_layer changes

* norm_layer and args fix

* Overwrite default dropout

* Remove docs from internal submodules.

* Fix tests

* Adding documentation.

* Link doc from main models.rst

* Fix min_temporal_size

* Adding crop/resize parameters in references script

* Release weights.

* Refactor dropout.

* Adding the weights table in the doc
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Vasilis Vryniotis <vvryniotis@fb.com>

* port `FixedSizeCrop` from detection references to prototype transforms

* mypy

* [skip ci] call invalid boxes and corresponding masks and labels

* cherry-pick missing functions from #6401

* fix feature wrapping

* add test

* mypy

* add input type restrictions

* add test for _get_params

* fix input checks

* Added ETH3D stereo dataset

* Small doc-reformating

* Removed assertions with no use, changed np conversion

* Added ETH3D stereo dataset

* Removed assertions with no use, changed np conversion

* rebased on main

* Revert "Removed assertions with no use, changed np conversion"

This reverts commit 1478a8c056bbed59b6f4a67f78a5cfac84cc9fda.

* Update to np.bool instead of np.bool_

* lint and mypy nit fix

* test nit

* expand has_any and has_all to also accept check callables

* add test and fix has_all

* add support for simple tensor images to CutMix, MixUp and RandomIoUCrop

* remove TODO

* remove pythonic syntax sugar

* simplify

* use concreate examples in test rather than abstract ones

* simplify further

* [proto] Ported RandomIoUCrop from detection refs

* Scope acceptable data types

* Added get_params test

* Added test__transform_empty_params

* Added support for OneHotLabel and tests

* Added tests for mask

* Updated error message

* Apply suggestions from code review
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Added support for OHE masks and tests

* Ignored mypy error

* Fixed forward call on sample

* Added a todo
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Added Falling Things datasets

* Renamed split to variant

* Update torchvision/datasets/_stereo_matching.py

Changed constant formatting
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* added SceneFLow variant datasets

* Changed split name to variant name

* removed trailing commented code line

* Added InStereo2k dataset

* Added Sintel Stereo dataset

* small refactor in tests

* Fixed doc formatting.

* candidate fix for FileNotFound on windows test

* Adressing comments

* Added Sintel Stereo dataset

* small refactor in tests

* Fixed doc formatting.

* candidate fix for FileNotFound on windows test

* Adressing comments

* rebased on main

* lint fix

* Added InStereo2k dataset

* added SceneFLow variant datasets

* Changed split name to variant name

* removed trailing commented code line

* Added Sintel Stereo dataset

* small refactor in tests

* Fixed doc formatting.

* candidate fix for FileNotFound on windows test

* Adressing comments

* Added Sintel Stereo dataset

* small refactor in tests

* Fixed doc formatting.

* candidate fix for FileNotFound on windows test

* Adressing comments

* rebased on main

* lint fix