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 MetalheadGetting 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
| Model Name | Constructor | Pre-trained? |
|---|---|---|
| UNet | UNet | N |