Metalhead.jlComputer vision models for Flux

联合创作 · 2023-09-26 01:16

Metalhead

Build Status

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.

Installation

]add Metalhead

Available models

Model Name Function Pre-trained?
VGG-11 VGG11 N
VGG-11 (w/ BN) VGG11 N
VGG-13 VGG13 N
VGG-13 (w/ BN) VGG13 N
VGG-16 VGG16 N
VGG-16 (w/ BN) VGG16 N
VGG-19 VGG19 Y
VGG-19 (w/ BN) VGG19 N
ResNet-18 ResNet18 N
ResNet-34 ResNet34 N
ResNet-50 ResNet50 Y
ResNet-101 ResNet101 N
ResNet-152 ResNet152 N
GoogLeNet GoogLeNet Y
Inception-v3 Inception3 N
SqueezeNet SqueezeNet Y
DenseNet-121 DenseNet121 Y
DenseNet-161 DenseNet161 N
DenseNet-169 DenseNet169 N
DenseNet-201 DenseNet201 N
浏览 4
点赞
评论
收藏
分享

手机扫一扫分享

编辑
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑
举报