GitLab

Co-authored-by: Federico Pozzi <federico.pozzi@argo.vision>

* Remove unnecessary PYTORCH_TEST_WITH_SLOW env

* Completely remove skip

* pre-download model weights in CI docs build

* move changes into template

* change docs image

Co-authored-by: Anton Thomma <anton@pri.co.nz>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* port Pad to prototype transforms

* use literal

* replace most asserts with exceptions

* fix formating issues

* fix linting and remove more asserts

* fix regresion

* fix regresion

* fix bug

* apply ufmt

* apply ufmt

* fix tests

* fix format

* fix None check

* fix detection models tests

* non scriptable any

* add more checks for None values

* fix retinanet test

* fix retinanet test

* Update references/classification/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update references/classification/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update references/optical_flow/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update references/optical_flow/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update references/optical_flow/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* make value checks more pythonic:

* Update references/optical_flow/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* make value checks more pythonic

* make more checks pythonic

* fix bug

* appy ufmt

* fix tracing issues

* fib typos

* fix lint

* remove unecessary f-strings

* fix bug

* Update torchvision/datasets/mnist.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/datasets/mnist.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/ops/boxes.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/ops/poolers.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/utils.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* address PR comments

* Update torchvision/io/_video_opt.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/models/detection/generalized_rcnn.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/models/feature_extraction.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/models/optical_flow/raft.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* address PR comments

* addressing further pr comments

* fix bug

* remove unecessary else

* apply ufmt

* last pr comment

* replace RuntimeErrors
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Added functional affine_bounding_box op with tests

* Updated comments and added another test case

* Update _geometry.py

* Fixed device mismatch issue
Added a cude/cpu test
Reduced the number of test samples

* Change optical flow train.py function name from validate to evaluate so it is similar to other references

* Add --device as parameter and enable to run in non distributed mode

* Format with ufmt

* Fix unneccessary param and bug

* Enable saving the optimizer and scheduler on the checkpoint

* Fix bug when evaluate before resume and save or load model without ddp

* Fix case where --train-dataset is None
Co-authored-by: Nicolas Hug <nicolashug@fb.com>

* refactor: port RandomHorizontalFlip to prototype API (#5523)

* refactor: merge HorizontalFlip and RandomHorizontalFlip

Add unit tests for RandomHorizontalFlip

* test: RandomHorizontalFlip with p=0

* refactor: remove type annotations from tests

* refactor: improve tests

* Update test/test_prototype_transforms.py
Co-authored-by: Federico Pozzi <federico.pozzi@argo.vision>
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* update latest stable version in README

* fix python version

* update python version

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

* Moving `sequence_to_str` to `torchvision._utils`

* Fix linter

* Rename test_prototype_utils test to test_internal_utils

* port RandomZoomOut from detection references to prototype transforms

* copy by default

* move padding parsing to stable API

* merge zoom out kernels into transform

* lint

* fix filling if no padding was done

* resolve todo

* [DEBUG] eurosat prototype on windows and macos

* print paths

* fix eurosat mock data setup

* revert changes

* minor cleanup

'fix' -> 'fig'

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

* revert changes to assert_equal

* add tolerances for batch vs single test

* basic stanford cars structure added.

* added class in the init.

* how to decode the images

* fix typo in init

* works partially

* formatting

* formatting

* formatting

* formatting

* formatting

* minor changes

* minor changes

* issue resolved

* both datapipes are in sync

* checksum dict keys changed

* formatting

* "ufmt formatting"

* "ufmt formatting"

* formatting

* minor change

* minor change

* test added

* lint (end of line issue)

* lint (end of line issue)

* minor changes

* mock dataset formatting

* minor changes

* hopefully final

* formatting

* prototype test

* prototype test

* prototype test

* prototype test

* prototype test

* prototype test

* prototype test

* prototype test

* Update builtin_dataset_mocks.py

* Update stanford_cars.py

* Update torchvision/prototype/datasets/_builtin/stanford_cars.py
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Update torchvision/prototype/datasets/_builtin/stanford_cars.py
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Update torchvision/prototype/datasets/_builtin/stanford_cars.py
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Update torchvision/prototype/datasets/_builtin/stanford_cars.py
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* formatting and cleaning

* formatting and cleaning

* Update torchvision/prototype/datasets/_builtin/stanford_cars.py
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* generate catefories script

* generate categories script added

* sigh

* sigh

* sigh

* minor changes

* linting
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* fix documentation version problems

* add missing import

* linting

* Measuring execution times of models.

* Speed up models by avoiding re-estimation of eager output

* Fixing linter

* Reduce input size for big models

* Speed up jit check method.

* Add simple jitscript fallback check for flaky models.

* Restore pytest filtering

* Fixing linter

* support grayscale / RGB alpha conversions

* use _max_valu from stable

* remove extra copy for PIL conversion

* simplify test image generation for color spaces with alpha channel

* use common _max_value in tests

* replace dynamically created dicts with if/else

* make color space conversion more explicit

* make even more explicit

* simplify alpha image generation

* fix if / elif

* add error for unknown conversions

* rename RGBA to RGB_ALPHA

* cleanup

* GRAYSCALE to GRAY

* Removed _max_value method and added a dictionary

Related to https://github.com/pytorch/vision/issues/5502

* Addressed failing tests and restored _max_value method

* Added xfailing test to switch quicker

* Switch to if/else impl

* Refactor preset transforms

* Making presets public.

* Adding improved MobileNetV2 weights

* Fix eval sizes.

* Update accuracy with batch-size 1 run

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

* port FiveCrop and TenCrop to prototype API

* fix ten crop for pil

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

* simplify implementation

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

* fix vanilla tensor image detection

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

* Fix accimage tests

* Adding workaround for accimage

* Refactoring

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

* Remove from models and references.

* Adding most tests and docs.

* Adding transforms tests.

* Remove unnecesary ipython notebook.

* Simplify tests.

* Addressing comments.