public
MobileNetv3
— struct
MobileNetv3(mode::Symbol = :small, width_mult::Number = 1; inchannels = 3, pretrain = false, nclasses = 1000)
Create a MobileNetv3 model with the specified configuration.
(reference).
Set pretrain = true
to load the model with pre-trained weights for ImageNet.
Arguments
mode
: :small or :large for the size of the model (see paper).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 channels in the input. The default value is 3.pretrain
: whether to load the pre-trained weights for ImageNetnclasses
: the number of output classes
See also Metalhead.mobilenetv3
.