Introduction

Tutorials

Developer guide

API Reference

public MobileNetv1struct

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 ImageNet
  • nclasses: The number of output classes

See also Metalhead.mobilenetv1.