- 21 Apr, 2022 1 commit
-
-
YosuaMichael authored
* Remove publication_year and interpolation meta * Add type to _COMMON_META and _COMMON_SWAG_META to prevent error from mypy check * Remove test to check interpolation and publication_year meta Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 22 Mar, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Moving basefiles outside of prototype and porting Alexnet, ConvNext, Densenet and EfficientNet. * Porting googlenet * Porting inception * Porting mnasnet * Porting mobilenetv2 * Porting mobilenetv3 * Porting regnet * Porting resnet * Porting shufflenetv2 * Porting squeezenet * Porting vgg * Porting vit * Fix docstrings * Fixing imports * Adding missing import * Fix mobilenet imports * Fix tests * Fix prototype tests * Exclude get_weight from models on test * Fix init files * Porting googlenet * Porting inception * porting mobilenetv2 * porting mobilenetv3 * porting resnet * porting shufflenetv2 * Fix test and linter * Fixing docs. * Porting Detection models (#5617) * fix inits * fix docs * Port faster_rcnn * Port fcos * Port keypoint_rcnn * Port mask_rcnn * Port retinanet * Port ssd * Port ssdlite * Fix linter * Fixing tests * Fixing tests * Fixing vgg test * Porting Optical Flow, Segmentation, Video models (#5619) * Porting raft * Porting video resnet * Porting deeplabv3 * Porting fcn and lraspp * Fixing the tests and linter * Porting docs, examples, tutorials and galleries (#5620) * Fix examples, tutorials and gallery * Update gallery/plot_optical_flow.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Fix import * Revert hardcoded normalization * fix uncommitted changes * Fix bug * Fix more bugs * Making resize optional for segmentation * Fixing preset * Fix mypy * Fixing documentation strings * Fix flake8 * minor refactoring Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Resolve conflict * Porting model tests (#5622) * Porting tests * Remove unnecessary variable * Fix linter * Move prototype to extended tests * Fix download models job * Update CI on Multiweight branch to use the new weight download approach (#5628) * port Pad to prototype transforms (#5621) * port Pad to prototype transforms * use literal * Bump up LibTorchvision version number for Podspec to release Cocoapods (#5624) Co-authored-by:
Anton Thomma <anton@pri.co.nz> Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * pre-download model weights in CI docs build (#5625) * pre-download model weights in CI docs build * move changes into template * change docs image * Regenerated config.yml Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
Anton Thomma <11010310+thommaa@users.noreply.github.com> Co-authored-by:
Anton Thomma <anton@pri.co.nz> * Porting reference scripts and updating presets (#5629) * Making _preset.py classes * Remove support of targets on presets. * Rewriting the video preset * Adding tests to check that the bundled transforms are JIT scriptable * Rename all presets from *Eval to *Inference * Minor refactoring * Remove --prototype and --pretrained from reference scripts * remove pretained_backbone refs * Corrections and simplifications * Fixing bug * Fixing linter * Fix flake8 * restore documentation example * minor fixes * fix optical flow missing param * Fixing commands * Adding weights_backbone support in detection and segmentation * Updating the commands for InceptionV3 * Setting `weights_backbone` to its fully BC value (#5653) * Replace default `weights_backbone=None` with its BC values. * Fixing tests * Fix linter * Update docs. * Update preprocessing on reference scripts. * Change qat/ptq to their full values. * Refactoring preprocessing * Fix video preset * No initialization on VGG if pretrained * Fix warning messages for backbone utils. * Adding star to all preset constructors. * Fix mypy. Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
Anton Thomma <11010310+thommaa@users.noreply.github.com> Co-authored-by:
Anton Thomma <anton@pri.co.nz>
-
- 15 Mar, 2022 1 commit
-
-
Joao Gomes authored
* replace most asserts with exceptions * fix formating issues * fix linting and remove more asserts * fix regresion * fix regresion * fix bug * apply ufmt * apply ufmt * fix tests * fix format * fix None check * fix detection models tests * non scriptable any * add more checks for None values * fix retinanet test * fix retinanet test * Update references/classification/transforms.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update references/classification/transforms.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update references/optical_flow/transforms.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update references/optical_flow/transforms.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update references/optical_flow/transforms.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * make value checks more pythonic: * Update references/optical_flow/transforms.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * make value checks more pythonic * make more checks pythonic * fix bug * appy ufmt * fix tracing issues * fib typos * fix lint * remove unecessary f-strings * fix bug * Update torchvision/datasets/mnist.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update torchvision/datasets/mnist.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update torchvision/ops/boxes.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update torchvision/ops/poolers.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update torchvision/utils.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * address PR comments * Update torchvision/io/_video_opt.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update torchvision/models/detection/generalized_rcnn.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update torchvision/models/feature_extraction.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update torchvision/models/optical_flow/raft.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * address PR comments * addressing further pr comments * fix bug * remove unecessary else * apply ufmt * last pr comment * replace RuntimeErrors Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com>
-
- 25 Feb, 2022 1 commit
-
-
Aditya Oke authored
* Add ops.conv3d * Refactor for conv2d and 3d * Refactor * Fix bug * Addres review * Fix bug * nit fix * Fix flake * Final fix * remove documentation * fix linter * Update all the implementations to use new Conv * Small doc fix Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Joao Gomes <jdsgomes@fb.com>
-
- 24 Feb, 2022 1 commit
-
-
Joao Gomes authored
-
- 09 Feb, 2022 1 commit
-
-
Aditya Oke authored
* Start adding example * Add thumbnail and text * Replace video * Improve * Change default weights of RAFT model builders * WIP * WIP * update handle_legacy_interface input * lots of stuff * Oops, wrong default * Typo * NITs * Reduce image size * Update gallery/plot_optical_flow.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Remove link to profile * Update gallery/plot_optical_flow.py Co-authored-by:
Aditya Oke <47158509+oke-aditya@users.noreply.github.com> * Address comments * Nits * Revert "Remove link to profile" This reverts commit 2c7a468c2bb0e14d4f39070524d216cbc153bf1d. Co-authored-by:
Nicolas Hug <nicolashug@fb.com> Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 07 Feb, 2022 1 commit
-
-
Nicolas Hug authored
* Change default weights of RAFT model builders * update handle_legacy_interface input * Oops, wrong default
-
- 02 Feb, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Move weight initialization in constructors. * Fixing mypy for ViT. * remove unnecessary import
-
- 27 Dec, 2021 1 commit
-
-
Aditya Oke authored
* Upgrade to py37 * Update
-
- 16 Dec, 2021 1 commit
-
-
Kai Zhang authored
* log API v3 * make torchscript happy * make torchscript happy * add missing logs to constructor * log ops C++ API as well * fix type hint * check function with isinstance Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 09 Dec, 2021 3 commits
-
-
Kai Zhang authored
* revamp log api usage method
-
Nicolas Hug authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Nicolas Hug authored
-
- 08 Dec, 2021 2 commits
-
-
Nicolas Hug authored
-
Nicolas Hug authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 07 Dec, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Adding logging calls for raft and vit * Linter fix
-
- 06 Dec, 2021 1 commit
-
-
Nicolas Hug authored
-