https://github.com/darsnack/metalhead.jl
Computer vision models for Flux
Science Score: 10.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
○CITATION.cff file
-
○codemeta.json file
-
○.zenodo.json file
-
○DOI references
-
✓Academic publication links
Links to: arxiv.org -
○Academic email domains
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (10.0%) to scientific vocabulary
Repository
Computer vision models for Flux
Basic Info
Statistics
- Stars: 0
- Watchers: 1
- Forks: 1
- Open Issues: 4
- Releases: 0
Metadata Files
README.md
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
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
| Model Name | Constructor | Pre-trained? |
|:-------------------------------------------------|:-----------------------------------------------------------------------------------------------|:------------:|
| AlexNet | AlexNet | N |
| ConvMixer | ConvMixer | N |
| ConvNeXt | ConvNeXt | N |
| DenseNet | DenseNet | N |
| EfficientNet | EfficientNet | N |
| EfficientNetv2 | EfficientNetv2 | N |
| gMLP | gMLP | N |
| GoogLeNet | GoogLeNet | N |
| Inception-v3 | Inceptionv3 | N |
| Inception-v4 | Inceptionv4 | N |
| InceptionResNet-v2 | InceptionResNetv2 | N |
| MLPMixer | MLPMixer | N |
| MobileNetv1 | MobileNetv1 | N |
| MobileNetv2 | MobileNetv2 | N |
| MobileNetv3 | MobileNetv3 | N |
| MNASNet | MNASNet | N |
| ResMLP | ResMLP | N |
| ResNet | ResNet | Y |
| ResNeXt | ResNeXt | Y |
| SqueezeNet | SqueezeNet | Y |
| Xception | Xception | N |
| WideResNet | WideResNet | Y |
| VGG | VGG | Y |
| Vision Transformer | ViT | Y |
Other Models
| Model Name | Constructor | Pre-trained? |
|:-------------------------------------------------|:-----------------------------------------------------------------------------------------------|:------------:|
| UNet | UNet | N |
Owner
- Name: Kyle Daruwalla
- Login: darsnack
- Kind: user
- Location: Cold Spring Harbor Lab, NY
- Website: darsnack.github.io
- Repositories: 67
- Profile: https://github.com/darsnack
NeuroAI scholar at CSHL