Introduction

Tutorials

API Reference

Metalhead

Dev CI Coverage

Metalhead.jl provides standard machine learning vision models for use with Flux.jl. The architectures in this package make use of pure Flux layers, and they represent the best-practices for creating modules like residual blocks, inception blocks, etc. in Flux. Metalhead also provides some building blocks for more complex models in the Layers module.

Installation

]add Metalhead

Available models

Model NameFunctionPre-trained?
VGGVGGN
ResNetResNetN
GoogLeNetGoogLeNetN
Inception-v3Inception3N
SqueezeNetSqueezeNetN
DenseNetDenseNetN
ResNeXtResNeXtN
MobileNetv1MobileNetv1N
MobileNetv2MobileNetv2N
MobileNetv3MobileNetv3N
MLPMixerMLPMixerN
ResMLPResMLPN
gMLPgMLPN
ViTViTN
ConvNeXtConvNeXtN
ConvMixerConvMixerN

Getting Started

You can find the Metalhead.jl getting started guide here.