EfficientNet family of models

EfficientNet(config::Symbol; pretrain::Bool = false, inchannels::Integer = 3,
             nclasses::Integer = 1000)

Create an EfficientNet model (reference).

Arguments

• config: size of the model. Can be one of [:b0, :b1, :b2, :b3, :b4, :b5, :b6, :b7, :b8].
• pretrain: set to true to load the pre-trained weights for ImageNet
• inchannels: number of input channels.
• nclasses: number of output classes.

See also Metalhead.efficientnet.

EfficientNetv2(config::Symbol; pretrain::Bool = false, inchannels::Integer = 3,
               nclasses::Integer = 1000)

Create an EfficientNetv2 model (reference).

Arguments

• config: size of the network (one of [:small, :medium, :large, :xlarge])
• pretrain: whether to load the pre-trained weights for ImageNet
• inchannels: number of input channels
• nclasses: number of output classes

See also efficientnet.