metadata
license: mit
MobileNet models ported from torchvision for use with Metalhead.jl. The script for creating this file can be found here.
To use this model in Julia, add the Metalhead.jl package to your environment. Then execute:
using Metalhead
# MobileNet V2 model
model = MobileNetv2(pretrain=true)
# MobileNet V3 models
model = MobileNetv3(:small, pretrain=true)
model = MobileNetv3(:large, pretrain=true)