GitLab

* update docstring for RandAugment

* update docstrings for pillow image

* Add RandomMixupCutmix.

* Add test with real data.

* Use dataloader and collate in the test.

* Making RandomMixupCutmix JIT scriptable.

* Move out label_smoothing and try roll instead of flip

* Adding mixup/cutmix in references script.

* Handle one-hot encoded target in accuracy.

* Add support of devices on tests.

* Separate Mixup from Cutmix.

* Add check for floats.

* Adding device on expect value.

* Remove hardcoded weights.

* One-hot only when necessary.

* Fix linter.

* Moving mixup and cutmix to references.

* Final code clean up.

* Fix RA bugs.

* Fix bins for TA.

* Initial Proposal

* Tensor Save Test + Test Name Fix

* Formatting + removing unused argument

* fix old argument

* fix isnan check error + indexing error with jit

* Fix Flake8 error.

* Fix MyPy error.

* Fix Flake8 error.

* Fix PyTorch JIT error in UnitTests due to type annotation.

* Fixing tests.

* Removing type ignore.

* Adding support of ta_wide in references.

* Move methods in classes.

* Moving new classes to the bottom.

* Specialize to TA to TAwide

* Add missing type

* Fixing lint

* Fix doc

* Fix search space of TrivialAugment.
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Vasilis Vryniotis <vvryniotis@fb.com>

* Adding docs for RandAugment.

* Fix docs.

* Adding randaugment implementation

* Refactoring.

* Adding num_magnitude_bins.

* Adding FIXME.

* fix

* add functional PIL typings

* fix types

* fix types

* fix a small one

* small fix

* fix type

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

* style: Added typing annotations

* style: Fixed typing

* style: Fixed typing

* Remove unnecessary any.

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

* style: Added missing typing annotations

* style: Fixed typing

* style: Fixed typing

* chore: Updated mypy.ini
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* WIP on backward op interpolation with AA

* Removed cuda tests and reformat cpp code

* Fixed clang wrong formatting

* Added channels last test case
Co-authored-by: vfdev-5 <vfdev-5@gmail.com>
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

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

* Moving tensors to the right device.

* Switch to gpu.medium

replaced byteTensor with from_numpy

fixed lint issues and copy related worning
Co-authored-by: Sofiane Abbar <sofa@fb.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Make crop work the same for pil and tensor

* Only call pad if needed in functional_tensor.crop

* Fix top-left functional_tensor.crop

* Update document for functional.crop

* Add other test cases of functional.crop

* Fix bug

* Fixing formattter

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

* Added support for bicubic mode with AA

* Updated comment in the test

* Fix inconsistent version pinning across yaml files.

* Pin pillow to >=5.3.0

* Pin pillow>=5.3.0 everywhere

* WIP Added antialias option to transforms.functional.resize

* Updates according to the review

* Excluded these C++ files for iOS build

* Added support for mixed downsampling/upsampling

* Fixed heap overflow caused by explicit loop unrolling

* Applied PR review suggestions
- used pytest parametrize instead unittest
- cast to scalar_t ptr
- removed interpolate aa files for ios/android keeping original cmake version

* Add docs for Interpolation modes

* Update torchvision/transforms/functional.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>

Co-authored-by: Nicolas Hug <nicolashug@fb.com>

* Fixed return docstrings

* Added some refs and corrected some parts

* more refs, and a note about dtypes
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* bump to 5.3.0

* Also make 3.6 CI runs rely on 5.3.0

* docs for resize

* address comment: describe antialiasing

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

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

* add _all for autoaugment

* adds docs

* add docs, test locally

* refactored as per code review

* 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