GitLab

* Modified the example to conform with the new DICT return.

* adding better stream documentation to examples

* Clearing up some documentation as a result of a feedback

* Formatting mostly

* Addressing Victor's comments.

* addressing fmassas comments

* remove unnecessary tab
Co-authored-by: Bruno Korbar <korbar@vggdev9.vggdev.cluster>
Co-authored-by: vfdev <vfdev.5@gmail.com>

* Fixing types.

* Dispatcher + Autocast.

* + Autograd.

* Formating.

* Clean up and refactor PSROIPool implementation:
- Remove primitive const declaration from method names.
- Using references when possible.
- Fix variable naming.

* Restore include headers.

* New line at end of file.

* Resolving conflict, final cleanup, ordering method consistently across files.

* Fixing types.

* Dispatcher + Autocast.

* + Autograd.

* Formating.

* Fixing return casting with autocast.

* Clean up and refactor ROIPool implementation:
- Remove primitive const declaration from method names.
- Using references when possible.

* Restore include headers.

* New line at end of file.

* Clean up and refactor ROIAlign implementation:
- Remove primitive const declaration from method names.
- Passing as const ref instead of value where possible.
- Remove unnecessary headers.

* Adding back include for cpu.

* Restore include headers.

* Fixing types.

* Dispatcher + Autocast.

* + Autograd.

* Clean up and refactor PSROIAlign implementation:
- Remove primitive const declaration from method names.
- Using references when possible.
- Sync naming of internal methods with other ops.

* Restoring names of internal methods to avoid conflicts.

* Restore include headers.

* Clean up and refactor DeformConv implementation:
- Remove primitive const declaration from method names.
- Passing as const ref instead of value where possible.
- Aligning method names between cpu and cuda.

* Adding newline.

* Adding back include for cpu.

* Restoring method names of private methods to avoid conflicts.

* Restore include headers.

* Change default eps value of FrozenBN.

* Update the unit-tests.`

* Update the expected values.

* Revert the expected value and use original eps=0 value for flaky tests.

* Post init change of eps.

* Styles.

* Clean up and refactor ROIAlign implementation:
- Remove primitive const declaration from method names.
- Remove unnecessary headers.
- Aligning method names between cpu and cuda.

* Adding back include for cpu.

* Restoring method names of private methods to avoid conflicts.

* Restore include headers.

- Replaced tuple creation by one acceptable by majority of compilers

* Splitting tuples of stride, padding and dilation of DeformConv.

* Fixing types.

* Dispatcher + Autocast.

* + Autograd.

* Moving contiguous() convertions away dispatcher and into the implementations.

* Removing rvalue references.

* style: Added annotation typing for vgg

* fix: Fixed annotation typing

* refactor: Removed un-necessary import

* fix: Added missing annotation for kwargs

* fix: Fixed constructor typing

* refactor: Refactored typing to minize changes

* refactor: Refactored typing cast

* fix: Fixed module list typing

* Minor refactoring based on static analysis on the code of models:
- Convert unnecessary value parameter to constant reference.
- Use move to avoid unnecessary copies.
- Eliminate unused include.

* Replace moves with const references.

* Fixing formatting.

* Remove explicit declaration on constructors.

- Remove unnecessary return.

- Replace deprecated string.h with cstring.
- Switch to nullptr.

* Convert to const reference and eliminate unnecessary bool casting.

* Removing unnecessary namespace use.

* Add float support to ColorJitter

* Fix byclass EMNIST

* Fix bymerge, balance, letters EMNIST

* Fix whitespace indent

* Revert unrelated file changes

* Revert unrelated file changes

* Removing unnecessary type conversions.

* Removing the transform and adding dummy class instead.
Co-authored-by: Vasileios Vryniotis <vvryniotis@fb.com>

* fix: Fixed constructor typing in models._utils

* fix: Fixed constructor typing in models.alexnet

* fix: Fixed constructor typing in models.mnasnet

* fix: Fixed constructor typing in models.squeezenet

* style: Added annotation typing for mmobilenet

* fix: Fixed type hinting of adaptive pooling

* refactor: Removed un-necessary import

* fix: Fixed constructor typing

* fix: Fixed list typing

* style: Added annotation typing for googlenet

* fix: Removed duplicate typing

* refactor: Moved factory function after class definition to fix typing

* fix: Fixed annotation typing

* refactor: Removed un-necessary import

* fix: Fixed typing

* refactor: Moved back up helper function and quote typed it

* style: Added annotation typing for inception

* refactor: Moved factory function after class definition

* style: Changed attribute setting for type hinting

* refactor: Removed un-necessary import

* fix: Fixed typing in constructors

* fix: Fixed kwargs typing

* style: Fixed lint

* refactor: Moved helpers function back and quote typed it

* style: Added annotation typing for resnet

* fix: Fixed annotation to pass classes

* fix: Fixed annotation typing

* fix: Fixed annotation typing

* fix: Fixed annotation typing for resnet

* refactor: Removed un-necessary import

* fix: Fixed constructor typing

* style: Added black formatting on _resnet

* style: Added annotation typing for squeezenet

* feat: Added typing for kwargs

* style: Added annotation typing for mnasnet

* refactor: Removed un-necessary import

* style: Added typing in models._utils

* fix: Removed non-necessary import

* fix: Removed type annotation of forward method

* refactor: Removed un-necessary import

* docs: Fixed docstring of RandomResizedCrop

* docs: Updated docstring
Co-authored-by: vfdev <vfdev.5@gmail.com>

* feat: Updated FrozenBN eps to align with BatchNorm

* feat: Added eps to __repr__ of FrozenBN

* test: Updated unittest of __repr__ for FrozenBN

* test: Updated unittest for eps value in BN and FrozenBN

* fix: Revert FrozenBN eps value

* test: Revert test on eps alignment between FrozenBN and BN

* feat: Added __repr__ to MultiScaleRoIAlign

* test: Added unittest for __repr__ of MultiScaleRoIAlign

* feat: Added feature map names in __repr__

* test: Updated unittest
Co-authored-by: vfdev <vfdev.5@gmail.com>

* Vectorize operations, across all feaure levels.

* Remove unnecessary other_outputs variable.

* Split per feature level.

* Perform batched_nms across feature levels.

* Add extra parameter for limiting detections before and after nms.

* Restoring default threshold.

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

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

* Ensure torchvision operators are registered in C++ via weak symbols

* Add note to README on how to ensure that torchvision operators are available in C++

* Fix dllimport/dllexport on windows, format files

* Factor out common macros in single file

* Expose cuda_version in the API, use it to avoid pruning of ops initializer

* Fix external DLL loading on wheels

* Use SetDefaultDllDirectoriess and AddDllDirectory

* Add previous paths

* Trigger debug

* Do not call SetDefaultDllDirectories.

* Do not call loadlibrary if the extensions were not compiled

* Fix lint issues

* [DEBUG] Check wheel relocation issues

* Call delocate on Mac

* Use yum instead of conda

* Do not copy ffmpeg dylibs

* Linux dry run

* Do not download FFmpeg on Mac

* Install bzip2

* Restore FFmpeg on Windows

* Remove ffmpeg temporarily

* Do not remove dependencies

* Disable FFmpeg temporarily on Linux wheels

* Test relocation in Linux

* Add docstring

* Call relocation script

* Minor error correction

* Import auditwheel only on Linux

* Restore ffmpeg again on Windows

* Return *device

* Try fix Windows

* Fix clang-format

* Start windows patchwork

* Copy all DLLs

* Disable FFmpeg on Windows for now

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