- 18 Jan, 2021 1 commit
-
-
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.
-
- 14 Jan, 2021 1 commit
-
-
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.
-