缓冲点问题和主线程使用 CoreML 被阻塞

Question

我正在使用 CoreML 删除给定图像中的背景，我有 2 个主要问题。

我在 Xcode 中嵌入的模型是 DeepLabV3.mlmodel。这是整个 class returns 最终图像：RemoveBackgroundCoreML_Final.swift 我刚刚导入到我的源代码中。它在完成工作方面运作良好。但是，我在第 107 行的 swift class 中有一个警告：

let data = Data(buffer: UnsafeBufferPointer(start: &cubeRGB, count: cubeRGB.count))

Initialization of 'UnsafeMutableBufferPointer' results in a dangling buffer pointer

更糟糕的情况是，根据我从 App Store 获取的某些设备的崩溃日志，这行代码使生产应用程序崩溃，即使我自己在开发环境中没有遇到任何崩溃。

这就是我从我的代码 Swift class 中调用 removeBackground 方法的方式：

let finalImage = image.removeBackground(returnResult: .finalImage)

另一个问题是，在 returns finalImage 行上方，但它也阻塞了应用程序中的主线程，其中有一些 UI 动画刚刚开始到运行之前显示一些过程（我怀疑这可能与缓冲区指针有关）。

我希望任何有 CoreML 或线程经验的人都可以帮助我，我非常感激。如果需要进一步说明，请询问。提前谢谢你。

Answer 1

由于您在创建指向它的指针后没有使用 cubeRGB，因此 Swift 优化器将在之后立即释放它。因为这是一个优化器的东西，你不会在调试模式下得到这个错误，只有在发布模式下。

Data(buffer:...) 将复制数组，这样很好。您只需要使数组保持活动状态，直到完成复制。这可以用 withExtendedLifetime 来完成，比如：

let data = withExtendedLifetime(cubeRGB) {
    return Data(buffer: UnsafeBufferPointer(start: &cubeRGB, count: cubeRGB.count))
}

我的语法可能有误，但类似这样。

缓冲点问题和主线程使用 CoreML 被阻塞

Buffer point issue and main thread is blocked using CoreML

multithreading

buffer

pointers

swift

coreml