- 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>
-
- 08 Feb, 2022 1 commit
-
-
Philip Meier authored
* properly deprecate legacy implementation * cleanup * use warning over deprecation directive * remove patch version * fix link in Kinetics docstring * Some more * fix affine functional tests Co-authored-by:Nicolas Hug <nicolashug@fb.com>
-
- 27 Jan, 2022 1 commit
-
-
Vasilis Vryniotis authored
-
- 06 Jan, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Improved meta-data for models. * Addressing comments from code-review. * Add parameter count. * Fix linter.
-
- 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 1 commit
-
-
Kai Zhang authored
* revamp log api usage method
-
- 22 Nov, 2021 1 commit
-
-
Kai Zhang authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 28 Oct, 2021 1 commit
-
-
Jirka Borovec authored
Co-authored-by:Nicolas Hug <nicolashug@fb.com>
-
- 19 Oct, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Move FCN methods to itsown package. * Fix lint. * Move LRASPP methods to their own package. * Move DeepLabV3 methods to their own package. * Adding deprecation warning for torchvision.models.segmentation.segmentation. * Refactoring deeplab. * Setting aux default to false. * Fixing imports. * Passing backbones instead of backbone names to builders. * Fixing mypy * Addressing review comments. * Correcting typing. * Restoring special handling for references.
-
- 07 Oct, 2021 1 commit
-
-
vfdev authored
* Use FX feature extractor for segm model * Removed use_fe option
-
- 04 Oct, 2021 1 commit
-
-
Philip Meier authored
* add ufmt as code formatter * cleanup * quote ufmt requirement * split imports into more groups * regenerate circleci config * fix CI * clarify local testing utils section * use ufmt pre-commit hook * split relative imports into local category * Revert "split relative imports into local category" This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51. * pin black and usort dependencies * fix local test utils detection * fix ufmt rev * add reference utils to local category * fix usort config * remove custom categories sorting * Run pre-commit without fixing flake8 * got a double import in merge Co-authored-by:Nicolas Hug <nicolashug@fb.com>
-
- 23 Aug, 2021 1 commit
-
-
F-G Fernandez authored
* style: Added typing annotations to segmentation/_utils * style: Added typing annotations to segmentation/segmentation * style: Added typing annotations to remaining segmentation models * style: Fixed typing of DeepLab * style: Fixed typing * fix: Fixed typing annotations & default values * Fixing python_type_check
-
- 17 Aug, 2021 1 commit
-
-
Vasilis Vryniotis authored
-
- 22 Jun, 2021 1 commit
-
-
Nicolas Hug authored
-
- 13 May, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Converting private parameters to public. * Add kwargs to handle extra params. * Add another kwargs. * Add arguments in _mobilenet_extractor.
-
- 29 Jan, 2021 2 commits
-
-
vfdev authored
Disable pretrained backbone downloading if pretrained is True Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Nicolas Hug authored
* Document undodcumented parameters * remove setup.cfg changes * Properly pass normalize down instead of deprecating it * Fix flake8 * Add new CI check * Fix type spec * Leave normalize be part of kwargs Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 27 Jan, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Making _segm_resnet() generic and reusable. * Adding fcn and deeplabv3 directly on mobilenetv3 backbone. * Adding tests for segmentation models. * Rename is_strided with _is_cn. * Add dilation support on MobileNetV3 for Segmentation. * Add Lite R-ASPP with MobileNetV3 backbone. * Add pretrained model weights. * Removing model fcn_mobilenet_v3_large. * Adding docs and imports. * Fixing typo and readme.
-
- 22 Dec, 2020 1 commit
-
-
Samuel Marks authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 12 Oct, 2020 1 commit
-
-
Adeel Hassan authored
Co-authored-by:vfdev <vfdev.5@gmail.com>
-
- 04 May, 2020 1 commit
-
-
Arash Javanmard authored
At the moment in the ASPP-Layer the number of output channels are predefined as a constant, which is good for DeepLab but not necessairly in other projects, where another out-channel Nr. is required. Also the number of "atrous rates" is fixed to three, which also could be sometimes more or less depending on the notwork-arch. Again these fixed values may make sense in DeepLab-Model but not necessarily in other type of models. This pull-req. contains the needed changes to make ASPP-Layer generic.
-
- 10 Apr, 2020 1 commit
-
-
moto authored
-
- 15 Nov, 2019 1 commit
-
-
eellison authored
-
- 20 Sep, 2019 1 commit
-
-
eellison authored
* script_fcn_resnet * Make old models load * DeepLabV3 also got torchscript-ready
-
- 11 Jun, 2019 1 commit
-
-
ekka authored
This PR uses a protected method for loading and initializing the segmentation models. Relevant #875
-
- 21 May, 2019 1 commit
-
-
Francisco Massa authored
Also adds documentation for the segmentation models
-
- 19 May, 2019 1 commit
-
-
Francisco Massa authored
* Move segmentation models to its own folder * Add missing files
-