Metalhead

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

julia> ]add Metalhead

Getting Started

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

Available models

To contribute new models, see our contributing docs.

Image Classification

Other Models