1. 27 Jan, 2021 1 commit
    • Vasilis Vryniotis's avatar
      Add MobileNetV3 architecture for Segmentation (#3276) · e2db2edd
      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.
      e2db2edd
  2. 18 Jan, 2021 1 commit
    • Vasilis Vryniotis's avatar
      Add MobileNetV3 architecture for Detection (#3253) · bf211dac
      Vasilis Vryniotis authored
      * Minor refactoring of a private method to make it reusuable.
      
      * Adding a FasterRCNN + MobileNetV3 with & w/o FPN models.
      
      * Reducing Resolution to 320-640 and anchor sizes to 16-256.
      
      * Increase anchor sizes.
      
      * Adding rpn score threshold param on the train script.
      
      * Adding trainable_backbone_layers param on the train script.
      
      * Adding rpn_score_thresh param directly in fasterrcnn_mobilenet_v3_large_fpn.
      
      * Remove fasterrcnn_mobilenet_v3_large prototype and update expected file.
      
      * Update documentation and adding weights.
      
      * Use buildin Identity.
      
      * Fix spelling.
      bf211dac
  3. 14 Jan, 2021 1 commit
    • Vasilis Vryniotis's avatar
      Add MobileNetV3 architecture for Classification (#3252) · 7bf6e7b1
      Vasilis Vryniotis authored
      * Add MobileNetV3 Architecture in TorchVision (#3182)
      
      * Adding implementation of network architecture
      
      * Adding rmsprop support on the train.py
      
      * Adding auto-augment and random-erase in the training scripts.
      
      * Adding support for reduced tail on MobileNetV3.
      
      * Tagging blocks with comments.
      
      * Adding documentation, pre-trained model URL and a minor refactoring.
      
      * Handling better untrained supported models.
      7bf6e7b1