GitLab

* Start adding types

* add typing

* Type prototype models

* fix optional type bug

* transient import

* Fix weights type

* fix import

* Apply suggestions from code review

Address nits
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Khushi Agrawal <khushiagrawal411@gmail.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* initial commit

* initial commit

* flake8

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

* Fixing minor issue on typing.

* Sample implementation for quantized resnet50.

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

We no longer cache the data in a custom binary, but simply read from the raw data directly.
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Annotate rpn

* Small fix

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

* Adding FasterRCNN ResNet50.

* Refactoring to remove duplicate code.

* Adding typing info.

* Setting weights_backbone=None as default value.

* Overwrite eps only for specific weights.

* Adding lightweight API for models.

* Adding resnet50.

* Fix preset

* Add fake categories.

* Fixing mypy.

* Add string=>weight conversion support on Enums.

* Temporarily hardcoding imagenet categories.

* Minor refactoring.

* Added Dropout parameter of Models

* Added argument description for dropout in MobileNet v2 and v3
Updated quantization/googlenet.py as per the changes in constructor in googlenet

* Moved new dropout parameter n the end

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

Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Add typing for imagelist, move params to docstring

* Add docs

* Start annotating utils

* checking

* Add annotations at _utils.py

* Remove unnecessary comments.

* re-checked typings

* Update typing

* Ignore small error

* Use optional tensor

* Ignore for JIT
Co-authored-by: Khushi Agrawal <khushiagrawal411@gmail.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

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

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

* mExplicitly copying array in pil_to_tensor

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

* Adding comments regarding implicit array deep copy in PILToTensor transform

* Update torchvision/transforms/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

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