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

标签 julia flux.jl flux-machine-learning

我有一个特定的计算机视觉问题,我想尝试使用一些预先训练的模型来解决。 Flux.jl 文档实际上并不像其他一些 ML 框架(以 PyTorch 为例)那样包含任何预先训练的模型。我如何在 Flux 中访问这些相关模型?

最佳答案

在 Flux 生态系统中,预训练计算机视觉模型等功能已被推断到名为 MetalHead.jl 的单独包中:https://github.com/FluxML/Metalhead.jl

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

julia> vgg19 = VGG19()
VGG19()

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

关于julia - 如何在 Flux.jl 中使用 VGG19?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/68336028/

相关文章:

types - 如何使用 Flux.jl 绘制函数及其梯度/导数

machine-learning - 使用 Flux.jl 进行逻辑回归

julia - 在 julia 中从 [1,2,0,0,4,0] 中查找非零元素的索引并用它们创建一个数组

methods - 如何向 Julia 中的现有函数添加方法?

julia - 由于策略争论,使用 NueralPDE.jl 中的 PhysicsInformedNN() 时出现 MethodError

machine-learning - 如何创建具有多个输出层的神经网络(Julia、Flux)

machine-learning - Julia 尺寸不匹配

performance - 为什么我的 julia 代码运行这么慢?

iterator - 在 Julia 中如何巧妙地处理零维度和迭代器