private
mobilenetv2
— function
mobilenetv2(width_mult, configs; max_width = 1280, nclasses = 1000)
Create a MobileNetv2 model. (reference).
Arguments
-
width_mult
: Controls the number of output feature maps in each block (with 1.0 being the default in the paper) -
configs
: A “list of tuples” configuration for each layer that details:t
: The expansion factor that controls the number of feature maps in the bottleneck layerc
: The number of output feature mapsn
: The number of times a block is repeateds
: The stride of the convolutional kernela
: The activation function used in the bottleneck layer
-
max_width
: The maximum number of feature maps in any layer of the network -
nclasses
: The number of output classes