如何在Flux.jl中使用VGG19？

Question

我有一个特定的计算机视觉问题，我想尝试使用一些预先训练的模型来解决。与其他ML框架(例如PyTorch)不同，Flux.jl文档实际上没有任何预先训练好的模型。我如何在Flux中访问这些相关的模型？

Answer 1

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

根据那里的文档，您可以通过执行以下操作来创建VGG19模型：

julia> vgg19 = VGG19()
VGG19()

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