julia - 如何在 Flux.jl 中使用 VGG19？

Question

我有一个特定的计算机视觉问题，我想尝试使用一些预先训练的模型来解决。Flux.jl 文档实际上并没有像其他一些 ML 框架（以 PyTorch 为例）那样的任何预训练模型。我将如何访问 Flux 中的那些相关模型？

Answer 1

在 Flux 生态系统中，诸如预训练计算机视觉模型之类的功能已被外推到一个名为 MetalHead.jl 的单独包中：https ://github.com/FluxML/Metalhead.jl

根据那里的文档，您可以通过以下方式创建 VGG19 模型：

julia> vgg19 = VGG19()
VGG19()

然后您可以将模型classify与输入图像一起传递给函数之类的东西以进行验证测试。