GitLab

This reverts commit 74de51d6.

* add base class for datasets tests

* add better type hints

* add documentation to subclasses

* add utility functions to create files / folders of random images and videos

* fix imports

* remove class properties

* fix smoke test

* fix type hints

* fix random size generation

* add Caltech256 as example

* add utility function to create grid of combinations

* add CIFAR100? as example

* lint

* add missing import

* improve documentation

* create 1 frame videos by default

* remove obsolete check

* return path of files created with utility functions

* [test] close PIL file handles before deletion

* fix video folder creation

* generalize file handle closing

* fix lazy imports

* add test for transforms

* fix explanation comment

* lint

* force load opened PIL images

* lint

* copy default config to avoid inplace modification

* enable additional arg forwarding

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

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

* restore cache

* Trigger CI again to see if cache is hit

- Added conda-forge channel

* Avoid freezing bn1 if all layers are trainable.

* Remove misleading comments.

Fixes #3393

* Replace type T with accumulator.

* Upcast tensors of box ops to avoid overflow in multiplications.

Summary:
In opt mode, the linker could strip out the symbols related to the torchvision ops if they had not been explicitly requested.

Adding link-whole is one way of fixing this.

Reviewed By: datumbox, ppwwyyxx

Differential Revision: D26249713

fbshipit-source-id: 4e55e2ebb483922fe308fd8b83456439bde728c9

* Fix test flakiness caused by rounding.

* Update test/test_transforms.py

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

Summary:
Pull Request resolved: https://github.com/pytorch/vision/pull/3370

Pull Request resolved: https://github.com/pytorch/vision/pull/3361

Convert pytorch_gpu to use GPU/RE

Reviewed By: datumbox

Differential Revision: D26264102

fbshipit-source-id: b45179215ecb1142f6468386c4fee1759dd82f0c

(cherry picked from commit 2f40a483d73018ae6e1488a484c5927f2b309969)
Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

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

* On MSVC, instruct the linker not to drop the _register_ops symbol

* Remove workaround for ops registration on windows
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

Reviewed By: zertosh

Differential Revision: D26396892

fbshipit-source-id: be0bd7fb54cb2f7bd0e0fb5222cc25b4930679a6

* remove generated dataset folders after download tests

* revert unnecessary style changes

Summary:
Noticed this copy-paste error while reading D26026259 (https://github.com/pytorch/vision/commit/f16322b596c7dc9e9d67d3b40907694f29e16357) for inspiration to solve a similar problem

Created from Diffusion's 'Open in Editor' feature.

Reviewed By: datumbox

Differential Revision: D26373056

fbshipit-source-id: d67f46243a280e8a59d12bad9dd730c1a3d512c9

* Adding the average_checkpoints() method.

* Adding the store_model_weights() method.

* Adding TODO placeholders.

* More placeholders.

* Add MobileNetV3 small pre-trained weights.

* Remove placeholders.

* aspect ratio must be a sampling from log scale.
reference from: https://github.com/pytorch/vision/blob/8317295c1d272e0ba7b2ce31e3fd2c048235fc73/torchvision/transforms/transforms.py#L833-L836



* add random erasing unittest code

* Increased threshold for difference in sampling rate

* move outside of the loop

* log_ratio move outside of the loop. RandomResizedCrop also
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* fix STL10 folds

* use np.int64 over np.long

* [android] android gradle project for ops

* Change CMakeLists to latest PyTorch

* Use mobilenet_v3 models for detection

Don't need to have two variants of the model anymore, but I'm not removing it for now

* Fix orientation when angle = 0

* [android][test_app] Fix YUV decoding

* Use smaller version of mobilenet model

* Divide inputs by 255 again

* [android] assets mobilenetv3
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

Summary:
This diff adds a new target to torchvision which enables users to use torchvision ops from C++.

For now, the `cpp_library` is not used by the `python_cpp_library`. We should instead refactor the logic in torchvision to directly use `cpp_library` instead.

There is currently an inconsistency between fbcode and OSS users. OSS users can import torchvision via
```
#include <torchvision/vision.h>
```
while fbcode users need to do
```
#include <torchvision/csrc/vision.h>
```
It would be good to fix this discrepancy in the future.

I didn't directly use `test_frcnn_tracing.cpp` due to complications for getting the `.pt` file in a way that works for both OSS and fbcode, so instead we added a self-contained test that should validate that the torchvision ops are properly registered and visible to JIT

Reviewed By: datumbox

Differential Revision: D26225669

fbshipit-source-id: 5dd9fb98dd58e854f95806e4860d02f54fc04ea4
Co-authored-by: Francisco Massa <fmassa@fb.com>

* extract some functionality from places365 fakedata for common use

* add a common DatasetTestcase

* add fakedata generation and tests for STL10

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

* make timeout a parameter

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

Summary:
We accept python bytes for video in existing API, such as `_read_video_timestamps_from_memory`.
We should do so to `_read_video_timestamps_from_memory` as well.

Reviewed By: fmassa

Differential Revision: D26216777

fbshipit-source-id: 5d7ecf15c5a65dc5c57f17b343e7c63b4933189e
Co-authored-by: Zhicheng Yan <zyan3@fb.com>

* kmnist

* emnist

* qmnist

* omniglot

* phototour

* sbdataset

* sbu

* semeion

* stl10

* svhn

* usps

* cifar100

* enable download logging for google drive

* celeba

* widerface

* lint

* add timeout logic

* lint

* debug CI connection to problematic server

* set timeout for ping

* [ci skip] remove ping

* revert debugging

* disable requests to problematic server

* re-enable all other tests
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Refactoring mobilenetv3 to make code reusable.

* Adding quantizable MobileNetV3 architecture.

* Fix bug on reference script.

* Moving documentation of quantized models in the right place.

* Update documentation.

* Workaround for loading correct weights of quant model.

* Update weight URL and readme.

* Adding eval.

* Check if the file is located on Google Drive

download_file() detects if the provided URL contains a link to Google
Drive and passes the request through the download_file_from_google_drive()
if it is the case.

* remove lazy re import

* add guard clause

* use urlparse

* add tests

* Fixing lint.

* make id matching more safe

* Fixing type check.

* Remove AnyStr.
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Vasilis Vryniotis <vvryniotis@fb.com>

* cleanup

* mnist

* lint
Co-authored-by: vfdev <vfdev.5@gmail.com>