GitLab

Summary: Enable analytics logging for MNIST and CIFAR

Reviewed By: colin2328, fmassa

Differential Revision: D29183132

fbshipit-source-id: 67d9f6881354f87659fdec4183b22a1be3ec0eb6
Co-authored-by: Dmitriy Genzel <dgenzel@fb.com>

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

* Fixed missing audio with video_reader backend

* Added unit test

* Adding a test for grayscale images.

* Doing the ops in torch, adding more checks.

* allow single-channel-images in draw_bounding_boxes

This is bug-fix for issue #4042.

https://github.com/pytorch/vision/issues/4042

* change made for passing linting requirements

#4042

* Initial commit

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

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

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

* replace pandas with system library to avoid crashes

* Lint

* Lint

* fixing unittest

* Minor comments removal

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

* remove asserts

* address pmeier formatting changes

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

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

* rename n_classes to num_classes

* formatting changes

* doc change to add ".mp4" to backported class

* formatting to correct line length

* adding **kwargs to Kinetics400 class

* remove urlib request and download the file directly

* annotations and files can be already downloaded

* test fix

* add download tests for Kinetics

* users now dont need to provide full path within the root for new Kinetics dataset

* linter

* Update test/test_datasets_download.py

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

* revert whitespace (3680#discussion_r626382842)

* addressing annotation_path parameter which is unnecessary

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

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

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

* expose num_download_workers as public

* swap os.isfile with check_integrity

* nit on private things

* special case if there are no default arguments

* revert changes to kinetics400 test case for BC

* add split_folder changes and support for legacy format

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

* pmeiers suggestions - root comment

* pmeiers comments - annotation attribute remmoved

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

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

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

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

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

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

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

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

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

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

* minor debugging

* nit picks

* only include public kwargs into defaults

* add _use_legacy_structure in favour of **kwargs

* add type hints for Kinetics400

* flake8

* flake8

* flake8

* rename to make thigs clearer

* permuting the output
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* Fixed some ffmpeg deprecation warnings in decoder

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

* Fixed comparison warnings in audio_stream and video_stream

* Fixed clang error

This reverts commit f6b6b510.

* fix

* add docs for VisionDataset

* add getitem

* add deprecated note to transform

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

* fix docs

* Update torchvision/datasets/vision.py

* flake8
Co-authored-by: Nicolas Hug <nicolashug@fb.com>
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

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

* Fixed read_image bug

* Removed unused import

* Skip tests if cv2 unavailable

* Removed cv2 dependency

* [doc] add minimum input size for alexnet builder

* [doc] add minimum input size for vgg builder

* [doc] add minimum input size for squeezenet builder

* [doc] add minimum input size for densenet builder

* [doc] add minimum input size for inception_v3 builder

* [doc] add minimum input size for googlenet builder
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* remove unused functions

* shut up annoying warnings via cast

* prabhat's comments

* Fixed clang format error
Co-authored-by: Prabhat Roy <prabhatroy@fb.com>

* Fixed audio-video synchronisation problem in read_video() when using  as unit

* Addressed review comments

* Added unit test

* Fix a bug when trainable_layers == 0

* Fix same issue on ssd.

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

* Moving tensors to the right device.

* Switch to gpu.medium

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