public
MobileNetv2
— struct
MobileNetv2(width_mult = 1.0; inchannels = 3, pretrain = false, nclasses = 1000)
Create a MobileNetv2 model with the specified configuration.
(reference).
Set pretrain
to true
to load the pretrained weights for ImageNet.
Arguments
width_mult
: Controls the number of output feature maps in each block (with 1.0 being the default in the paper; this is usually a value between 0.1 and 1.4)inchannels
: The number of input channels. The default value is 3.pretrain
: Whether to load the pre-trained weights for ImageNetnclasses
: The number of output classes
See also Metalhead.mobilenetv2
.