如何在 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 函数以及用于验证测试的输入图像。