如何在 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 的函数以进行验证测试。