GitLab

* Use List[int] instead of int for window_size and shift_size

* Make PatchMerging and SwinTransformerBlock able to handle 2d and 3d cases

* Separate patch embedding from SwinTransformer and enable to get model without head by specifying num_heads=None

* Dont use if before padding so it is fx friendly

* Put the handling on window_size edge cases on separate function and wrap with torch.fx.wrap so it is excluded from tracing

* Update the weight url to the converted weight with new structure

* Update the accuracy of swin_transformer

* Change assert to Exception and nit

* Make num_classes optional

* Add typing output for _fix_window_and_shift_size function

* init head to None to make it jit scriptable

* Revert the change to make num_classes optional

* Revert unneccesarry changes that might be risky

* Remove self.head declaration

* add tests for F.pad_bounding_box

* Added correctness tests for pad and reimplemented bbox op to keep dtype

* Update _geometry.py
Co-authored-by: vfdev <vfdev.5@gmail.com>

Add .float() before .mean() on test_backbone_utils.py because .mean() dont accept integer dtype (#6090)

* Validate against expected files on videos

* Plus tests for autocast

* feat: add functional center crop on mask

* test: add correctness center crop with random segmentation mask

* test: improvements

* test: improvements

* Apply suggestions from code review
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Federico Pozzi <federico.pozzi@argo.vision>
Co-authored-by: vfdev <vfdev.5@gmail.com>
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Indicate download parameter is deprecated and will be removed

* change default to None and raise warning if user isnt using default

* aonfajoajgfnjaognojag

* Remove `(N, T, H, W, C) => (N, T, C, H, W)` conversion on presets

* Update docs.

* Fix the tests

* Use `output_format` for `read_video()`

* Use `output_format` for `Kinetics()`

* Adding input descriptions on presets

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

Throw warning for empty masks or box tensors on draw_segmentation_masks and draw_bounding_boxes (#5857)

* Fixing the IndexError in draw_segmentation_masks

* fixing the bug on draw_bounding_boxes

* Changing fstring to normal string

* Removing unecessary conversion

* Adding test for the change

* Adding a test for draw seqmentation mask

* Fixing small mistake

* Fixing an error in the tests

* removing useless imports

* ufmt
Co-authored-by: LEGRAND Matthieu <legrand.ma@chu-toulouse.fr>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Dataset creation now supports "new" version of Kinetics dataset

* remove unnecessary warning for now

* provide kinetics option

* new reading somehow doesn't need BHWC to BCHW transform

* Addressing minor comments

* Adding kinetics deprication warning for the old Kinetics400 class

* lint error

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

* Updating README

* Remove BHWC to BCHW

* Put warning back

* formatting
Co-authored-by: Bruno Korbar <bkorbar@quansight.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* improve error handling for GDrive downloads

* perform HTML check regardless of MD5 check

* Deactivate CelebA download

* flake8

* Do proto version

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

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

* To resolve issue #5964

Add note for resnet architecture

* Update resnet.py

* Update resnet.py

* Update resnet.rst

* Fix stylings

* Add the same notes on model builders

* Improve description

* Apply the change everywhere

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

* Adding an MLP block.

* Adding documentation

* Update typos.

* Fix inplace for Dropout.

* Apply recommendations from code review.

* Making changes on pre-trained models.

* Fix linter

* Add swin on hubconfig.

* Add swin b/s in the `slow_models` list.

* add swin_s and swin_b variants

* fix swin_b params

* fix n parameters and acc numbers

* adding missing acc numbers

* apply ufmt

* Updating `_docs` to reflect training recipe

* Fix exted for swin_b
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Remove models.rst

* Remove '- New'

* Put back torchhub section where it originally was

* Doc revamp for optical flow models

* Some more

* Classif models

* Detection

* Segmentation

* quantization

* Video

* optical flow

* tests

* Fix docs

* Fix Video dataset

* Consistency for RAFT dataset names

* use ImageNet-1K

* Use COCO-val2017-VOC-labels for segmentation

* formatting

* Replace jcenter for android

* Removing revision 0 from version

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

* Adding docs for quantized models.

* Adding docs for video models.

* Adding docs for segmentation models.

* Adding docs for optical flow models.

* Adding docs for detection models.

* Fix typo.

* Make changes from code-review.

* Cleanup ops

* Address nits
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* try re arranging docs

* rewrite docs a bit

* Apply suggestions from code review
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Improving the auto-gen doc.

* Adding details for AlexNet, ConvNext, DenseNet, EfficientNets, GoogLeNet and InceptionV3.

* Fixing location of `_docs`

* Adding docs in the remaining classification models.

* Fix linter

* update paper link for FCOS refence

* remove 'updated version'
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* simplify OnlineResource.load

* [PoC] merge mock data preparation and loading

* Revert "cache mock data based on config"

This reverts commit 5ed6eedef74865e0baa746a375d5ec1f0ab1bde7.

* Revert "[PoC] merge mock data preparation and loading"

This reverts commit d62747962f9ed6a7b0b80849e7c971efabb5d3da.

* remove preprocess returning a new path in favor of querying twice

* address test comments

* clarify comment

* mypy

* use builtin decompress utility

* merge mock data preparation and loading

* address comments

* fix extra file creation

* remove tmp folder

* inline images meta creation in coco mock data

* Update the preprocessing decription for RAFT.

* Fixing incorrect usage of models.

* Fixing the content of viz utils

* Addressing review comments

* docs: Added quantized ResNeXt to new docs

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

* Minor updates on model examples.

* Improving wording of auto-generated docs.

* Add general info for pre-trained weights.

* Updating torch hub

* Minor updates

* Make lengthy meta-data partially visible

* Adding meta-data and reference info.

* Minor corrections

* Update docs/source/models_new.rst
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Moving Torch hub section at the end
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Requested here https://github.com/pytorch/vision/pull/5898#discussion_r864765799

.

* Fix tests

* ufmt, not black
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Add weight for mnasnet0_75 and mnasnet1_3

* Fix missing comma

* Add PR url as recipe, and update the metrics

* Add weights to legacy handler

* Update docs to specify there are weights available

* Prefixing `_get_enum_from_fn` with underscore

* Exposing `get_weight` to Torch Hub