我有一个特定的计算机视觉问题,我想尝试使用一些预先训练的模型来解决。 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/