• 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
test_models_detection_utils.py 3.2 KB