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

如何解决缓冲点问题和主线程被 CoreML 阻塞

我正在使用 CoreML 删除给定图像中的背景，但我遇到了两个主要问题。

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

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

'UnsafeMutableBufferPointer' 的初始化导致悬空缓冲区指针

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

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

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

另一个问题是，上面的行返回 finalImage 但它也阻塞了应用程序中的主线程，该线程有一些 UI 动画之前刚刚开始运行以显示一些进程（我怀疑这可能与缓冲区指针有关).

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

解决方法

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

Data(buffer:...) 会复制数组，这样很好。您只需要让数组保持活动状态，直到完成复制。这可以通过 withExtendedLifetime 完成，例如：

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

我可能有语法错误，但类似的东西。