GitLab

* add benchmark utility for datasets

* cleanup

* cleanup

* add benchmarks for MNIST and variants

* simplifiy return if exeception

* refactor benchmark instantiation logic
Co-authored-by: Prabhat Roy <prabhatroy@fb.com>

* Use FX feature extractor for segm model

* Removed use_fe option

Co-authored-by: Prabhat Roy <prabhatroy@fb.com>

Co-authored-by: Prabhat Roy <prabhatroy@fb.com>

Co-authored-by: Prabhat Roy <prabhatroy@fb.com>

* add prototype datasets for MNIST and variants

* fix mypy

* fix EMNIST labels

* fix code format

* avoid encoding + decoding in every step

* discard data at the binary level instead of after decoding

* cleanup

* fix mypy

* add prototype dataset for CelebA

* fix code format

* fix mypy

* hardcode fmtparams

* fix mypy

* replace KeyZipper with Zipper for annotations

* add prototype for SBD dataset

* cleanup

* add prototype for VOC dataset

* fix code format

* fix mypy

* resolve todo

* add prototype for CIFAR datasets

Conflicts:
	torchvision/prototype/datasets/_builtin/__init__.py
	torchvision/prototype/datasets/utils/_internal.py

* fix mypy

* cleanup

* more cleanup

* revert unrelated changes

* fix code format

* avoid decoding twice by default

* revert unrelated change

* cleanup

* Add str param

* Update test to include str

* Fix mypy

* Remove a small bracket

* Test more robustly

* Update docstring and test:

* Apply suggestions from code review
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/utils.py

Small docstring fix

* Update torchvision/utils.py

* remove unnecessary renaming
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
Co-authored-by: Nicolas Hug <nicolashug@fb.com>

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

* Allow redefinition for mypy

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

* add best weights and x_1_6, x_3_2, y_1_6, y_3_2, y_32 weights

* add best weights and x_1_6, x_3_2, y_1_6, y_3_2, y_32 weights

* add weights for x_16gf, x_32gf, y_16gf

* add ufmt as code formatter

* cleanup

* quote ufmt requirement

* split imports into more groups

* regenerate circleci config

* fix CI

* clarify local testing utils section

* use ufmt pre-commit hook

* split relative imports into local category

* Revert "split relative imports into local category"

This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51.

* pin black and usort dependencies

* fix local test utils detection

* fix ufmt rev

* add reference utils to local category

* fix usort config

* remove custom categories sorting

* Run pre-commit without fixing flake8

* got a double import in merge
Co-authored-by: Nicolas Hug <nicolashug@fb.com>

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

* draft commit

* Polish and add corresponding test

* Update docs

* Update torchvision/models/feature_extraction.py

* Update docs/source/feature_extraction.rst
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* add prototype for `Caltech256` dataset

* silence mypy

* add torchdata as prototype dependency

* ignore torchdata for mypy

* move git dependency out of conda environment file

* try unsetting PYTORCH_VERSION env var

* cleanup

* move installation of torchdata after installtion of PyTorch

* regenerate circlci config

* remove branch from torchdata download link
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* Moving _make_divisible to utils.

* Replace the old ConvBNReLU and ConvBNActivation layers

* Fix minor bug.

* Moving SE layer to ops.

* Adding deprecation warnings on old layers.

* Apply changes to regnets.

* initial code

* add SqueezeExcitation

* initial code

* add SqueezeExcitation

* add SqueezeExcitation

* regnet blocks, stems and model definition

* nit

* add fc layer

* use Callable instead of Enum for block, stem and activation

* add regnet_x and regnet_y model build functions, add docs

* remove unused depth

* use BN/activation constructor and ConvBNActivation

* add expected test pkl files

* allow custom activation in SqueezeExcitation

* use ReLU as the default activation

* initial code

* add SqueezeExcitation

* initial code

* add SqueezeExcitation

* add SqueezeExcitation

* regnet blocks, stems and model definition

* nit

* add fc layer

* use Callable instead of Enum for block, stem and activation

* add regnet_x and regnet_y model build functions, add docs

* remove unused depth

* use BN/activation constructor and ConvBNActivation

* reuse SqueezeExcitation from efficientnet

* refactor RegNetParams into BlockParams

* use nn.init, replace np with torch

* update README

* construct model with stem, block, classifier instances

* Revert "construct model with stem, block, classifier instances"

This reverts commit 850f5f3ed01a2a9b36fcbf8405afd6e41d2e58ef.

* remove unused blocks

* support scaled model

* fuse into ConvBNActivation

* make reset_parameters private

* fix type errors

* fix for unit test

* add pretrained weights for 6 variant models, update docs

* Reuse EfficientNet SE layer.

* Deprecating the mobilenetv3.SqueezeExcitation layer.

* Passing the right activation on quantization.

* Making strict named param.

* Set default params if missing.

* Fixing typos.

* cleanup image folder

* make shuffling mandatory

* rename parameter in home() function

* don't show builtin list

* make categories optional in dataset info

* use pseudo-infinite buffer size for shuffler
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* update channels parameter to every calling to check_functional_vs_PIL_vs_scripted

* update adjust_saturation

* update docstrings for functional transformations

* parametrize channels

* update docstring of ColorJitter class

* move channels to class's parameter

* remove testing channels for geometric transforms

* revert redundant changes

* revert redundant changes

* update grayscale test cases for randaugment, autoaugment, trivialaugment

* update docstrings of randaugment, autoaugment, trivialaugment

* update docstring of ColorJitter

* fix adjust_hue's docstring

* change test equal tolerance

* refactor grayscale tests

* make get_grayscale_test_image private

* Update gpu atomics include path

THC is being removed, and this header now lives in ATen.
See pytorch/pytorch#65470.

* Empty commit to trigger build
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* add API for new style datasets

* cleanup
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

Co-authored-by: Nicolas Hug <nicolashug@fb.com>

* add prototype image folder dataset

* remove decoder datapipe

* [PROPOSAL] add RandomPicker

* refactor data loading

* fix mypy

* remove per-category datapipes

* fix mypy
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* update docstring for RandAugment

* update docstrings for pillow image

* ops.masks_to_bounding_boxes

* test fixtures

* unit test

* ignore lint e201 and e202 for in-lined matrix

* ignore e121 and e241 linting rules for in-lined matrix

* draft gallery example text

* removed type annotations from pytest fixtures

* inlined fixture

* renamed masks_to_bounding_boxes to masks_to_boxes

* reformat inline array

* import cleanup

* moved masks_to_boxes into boxes module

* docstring cleanup

* updated docstring

* fix formatting issue

* gallery example

* use torch

* use torch

* use torch

* use torch

* updated docs and test

* cleanup

* updated import

* use torch

* Update gallery/plot_repurposing_annotations.py
Co-authored-by: Aditya Oke <47158509+oke-aditya@users.noreply.github.com>

* Update gallery/plot_repurposing_annotations.py
Co-authored-by: Aditya Oke <47158509+oke-aditya@users.noreply.github.com>

* Update gallery/plot_repurposing_annotations.py
Co-authored-by: Aditya Oke <47158509+oke-aditya@users.noreply.github.com>

* Autodoc

* use torch instead of numpy in tests

* fix build_docs failure

* Closing quotes.
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Aditya Oke <47158509+oke-aditya@users.noreply.github.com>

* allow custom activation in SqueezeExcitation

* use ReLU as the default activation

* make scale activation parameterizable
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>