为什么在 IOS 上运行 torchscript 模型推理会导致线程错误？

编程问答 2022-04-21

问题描述

我一直在尝试将在python上开发的pytorch模型集成到IOS中。我看过的例子来自这个github repo。

我在自己的应用程序中使用了相同的 d2go 模型。我注意到的一件事是，如果模型推理代码没有包含在 dispatchQueue 全局中，如下所示

dispatchQueue.global().async {
    guard let outputs = self.inferencer.module.detect(image: &pixelBuffer) else {
        return
    }

我收到类似 Thread 1: EXC_BAD_ACCESS (code=1,address=0x7ffeeb4e0000) 的错误，或者如果我的模型运行推理所需的时间太长，即使它包含在上面的 dispatchQueue 代码中，我也会收到类似 Thread 4: EXC_BAD_ACCESS (code=1,address=0x7ff159bed010) 的错误。

我不确定线程​​在这种情况下是如何工作的。在新的 SwiftUI 框架中按下按钮时，我正在运行代码。

关于为什么会发生这种情况的任何直觉？我已经在模拟器上尝试过上述方法

解决方法

您可能应该在同一范围内（在调度块内）声明“pixelBuffer”

dispatch-queueiosiosmultithreadingmultithreadingpytorchtorchscript