|
|
--- |
|
|
license: mit |
|
|
--- |
|
|
|
|
|
MobileNet models ported from [torchvision](https://pytorch.org/vision/stable/index.html) for use with [Metalhead.jl](https://github.com/FluxML/Metalhead.jl). The script for creating this file can be found [here](https://github.com/FluxML/Metalhead.jl/blob/master/scripts/port_torchvision.jl). |
|
|
|
|
|
To use this model in Julia, [add the Metalhead.jl package to your environment](https://pkgdocs.julialang.org/v1/managing-packages/#Adding-packages). Then execute: |
|
|
```julia |
|
|
using Metalhead |
|
|
|
|
|
# MobileNet V2 model |
|
|
model = MobileNetv2(pretrain=true) |
|
|
|
|
|
# MobileNet V3 models |
|
|
model = MobileNetv3(:small, pretrain=true) |
|
|
model = MobileNetv3(:large, pretrain=true) |
|
|
``` |
|
|
|
