• Vasilis Vryniotis's avatar
    Port Multi-weight support from prototype to main (#5618) · 11bd2eaa
    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: default avatarNicolas 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: default avatarNicolas 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: default avatarAnton Thomma <anton@pri.co.nz>
    Co-authored-by: default avatarVasilis 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: default avatarPhilip Meier <github.pmeier@posteo.de>
    Co-authored-by: default avatarAnton Thomma <11010310+thommaa@users.noreply.github.com>
    Co-authored-by: default avatarAnton 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: default avatarNicolas Hug <contact@nicolas-hug.com>
    Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
    Co-authored-by: default avatarAnton Thomma <11010310+thommaa@users.noreply.github.com>
    Co-authored-by: default avatarAnton Thomma <anton@pri.co.nz>
    11bd2eaa
plot_visualization_utils.py 19.1 KB