GitLab

* Add quantized version of nms

* Added tests

* Compute areas only once

* remove calls to dequantize_val

* fix return type for empty tensor

* flake8

* remove use of scale as it gets cancelled out

* simpler int convertion in tests

* explicitly set ovr to double

* add tests for more values of scale and zero_point

* comment about underflow

* remove unnecessary accessor

* properly convert to float for division

* Add comments about underflow

* explicitely cast coordinates to float to allow vectorization

* clang

* clang  again

* hopefully OK now

* multithreading allowed in stream codec context

* numThreads is passed as a decoder parameter. At this stage code should be unchanged

* enabling multithreading in videoReader API

* moving defaults to header files

* replace long with int64_t because torchscript

* docstring for Num threads

* Enable codec-related heuristics as defaults

* Update torchvision/csrc/io/decoder/stream.cpp
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Fixing build errors

* minor docs

* Linting

* updating defaults for the C++ function calls to be single threaded

* adding special case for single threaded stuff
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* factor out find_classes

* use find_classes in video datasets

* adapt old tests

* add draw segm masks

* rewrites with new api

* fix flaky colors

* fix resize bug

* resize for sanity

* cleanup

* project the image

* Minor refactor to adopt num classes

* add uint8 in docstring

* adds alpha and docstring

* move code a bit down

* Minor fix

* fix type check

* Fixing resize bug.

* Fix type of alpha.

* Remove unnecessary RGBA conversions.

* update docs to supported only rgb

* minor edits

* adds tests

* shifts masks up

* change tests and impelementation for bool

* change mode to L

* convert to float

* fixes docs
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Vasilis Vryniotis <vvryniotis@fb.com>
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

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

* added .item() for target inside getitem

* changed fakedataset's expected return type
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

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

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

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

* remove caching from (Fashion|K)?MNIST

* remove unnecessary lazy import

* remove false check of binaries against the md5 of archives

* remove caching from EMNIST

* remove caching from QMNIST

* lint

* fix EMNIST

* streamline QMNIST download

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

* generalize extract_archive

* [test] re-enable extraction tests on windows

* add tests for detect_file_type

* add error messages to detect_file_type

* Revert "[test] re-enable extraction tests on windows"

This reverts commit 7fafebb0f6b4c49bd72c4b5e0a0b4b8c432bce57.

* add utility functions for better mock call checking

* add tests for decompress

* simplify logic by using pathlib

* lint

* Apply suggestions from code review
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* make decompress private

* remove unnecessary checks

* add error message

* fix mocking

* add remaining tests

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

* use head request for redirects

* remove requests dependency

* add tests

* fix bug

* remove tests

* fix comment

* fix comment

* add warning

* fix syntax error

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

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

* add missing docs

* tries fixing docs

* fixes docs

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

* add mirrors to (Fashion|K)MNIST

* fix download tests for MNIST

We are experiencing 403s when trying to download from the main mnist
site so lets fallback to our own mirror on failure.
Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

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

* add _all for autoaugment

* adds docs

* add docs, test locally

* refactored as per code review

* Stash commit

* Refactoring the current tests in a simpler file

* Addressing Francisco's comments, attempt 1

* Remove old tests

* audio changes

* adding some comments

* Stash commit

* Refactoring the current tests in a simpler file

* Addressing Francisco's comments, attempt 1

* Remove old tests

* audio changes

* adding some comments

* adding the py39 check stuff
Co-authored-by: Bruno Korbar <korbar@vggdev9.vggdev.cluster>
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* use bincount instead of hist

* only use bincount when on CPU

* Added equality test for CPU vs cuda

* Fix flake8 and tests

* tuple instead of int for size

* new batched_nms implem

* flake8

* hopefully fix torchscipt tests

* Use where instead of nonzero

* Use same threshold (4k) for CPU and GPU

* Remove use of argsort

* use views again

* remove print

* trying stuff, I don't know what's going on

* previous passed onnx checks so the error isn't in _vanilla func. Trying to return vanilla now

* add tracing decorators

* cleanup

* wip

* ignore new path with ONNX

* use vanilla if tracing...????

* Remove script_if_tracing decorator as it was conflicting with _is_tracing

* flake8

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

* WIP, still needs tests and docs

* tests

* flake8

* Docs + fixed some tests

* proper error messages

* add custom user agent for download_url

* fix progress bar

* lint

* [test] use repo instead of nightly for download tests

* fixed origin head

* fixed inconsistent casting

* updated functional_tensor.py

Modified the .to() method to convert_image_dtype() method.

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

* add tests for PhotoTour dataset

* fix grayscale image generation

* fix test_feature_types for a examples of a single feature

* make image size variable instead of hard coding it

* make dataset length variable instead of hard coding it

* replace numpy with torch

* fix typo

* include intended structure of root directory in docstring of Kinetics400

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

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

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

Current implementation is generating bad graph after onnx conversion. So replacing with flatten like in mobilenetv3 code.
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* deprecate _video modules

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

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