在使用 Flutter 的 Android 应用中使用 Pytorch LightGlue

Question

我很难让 LightGlue 在 Android 上运行。我认为因为它针对速度进行了优化，所以它将是尝试了解有关构建电话应用程序的更多信息的完美目标，但我无法让它工作。

我只找到了如何让 torchscript 模型在 Android 上运行的资源，但没有找到纯 pytorch 模型。所以我尝试将模型转换为torchscript，但没有成功。 我试过：

1. torch.jit.trace(model, example)

   ，但模型似乎严重依赖于输入数据，因此当我使用不同的输入时，跟踪它就会中断。

2. torch.jit.script(model)

   但无法让它工作，因为似乎每隔一行都使用与脚本功能不兼容的功能

3. torch.compile(model)

   用于编译，但我不知道如何保存或加载编译后的模型。所以这个也行不通

我不依赖flutter，所以如果你能推荐另一个框架，我很乐意使用它。

有什么想法可以让它发挥作用吗？

Answer 1

我和你有同样的问题。你现在解决这个问题了吗？