• Shixin's avatar
    [MobileNet] Add Mobilenet Backbone Implementation (#9303) · e61588cd
    Shixin authored
    
    
    * factor make_divisible function and move round_filters to nn_layers
    
    * modify SqueezeExcitation to add two additional parameter: divisible_by and gating_activation
    
    * modify the InvertedBottleneckBlock to include 1. use_depthwise, 2. use_residual, 3. regularize_depthwise additional boolean flag; Add control for depthwise activation and regularizer; remove expand_ratio from SqueezeExcitation
    
    * add Conv2DBNBlock definition
    
    * add mobilenet v2, v3 implementation
    
    * add mobilenet v1
    
    * put mobilenet_base into class body
    
    * fix a type hint error
    
    * the invertedbottlenetblock is different for mobilenet and efficientnet. Made necessary changes to cope both.
    
    * add target_backbone while call invertedbottleneckblock
    
    * add relu6 and hard_sigmoid
    
    * add test for mobilenet
    
    * add mobilenet to factory
    
    * fix some typo; link the reference to the architectures
    
    * remove future import
    Co-authored-by: default avatarShixin Luo <luoshixin@google.com>
    e61588cd
tf_utils.py 5.73 KB