public
MobileNetv1
— struct
MobileNetv1(width_mult = 1; inchannels = 3, pretrain = false, nclasses = 1000)
Create a MobileNetv1 model with the baseline 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.mobilenetv1
.