Android 上的 TensorFlow Lite PoseNet 因内存问题而崩溃

Question

我正在尝试创建一个使用 TensorFlow Lite PoseNet 进行人体姿势估计的 Android 应用。 我遇到的问题是本机内存缓慢增加，直到崩溃。 即使我运行demo app，它也会在大约 20 分钟后在我的 S10 上崩溃。 我尝试对其进行分析，但我认为这不是泄漏，因为如果我对其进行编码以便解释器中断，那么垃圾收集就能够跟上。

我想让它以大约每秒 15 次的速度进行估计，这在几分钟内似乎做得很好。有没有办法调整它以使其运行更长时间，或者在三星 S10 等设备上运行是否不切实际？

Answer 1

Android PoseNet 演示应用中存在内存泄漏，除非您启用window.addFlags(WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON)，否则不会引起注意

PosenetActivity.kt captureSession!!.setRepeatingRequest 正在使用一个 backgroundHandler，它持有一个阻止本机内存被清理的引用。 setRepeatingRequest 不需要回调和处理程序。 改变这个

      previewRequest = previewRequestBuilder!!.build()
      captureSession!!.setRepeatingRequest(
        previewRequest!!,
        captureCallback, backgroundHandler
      )

到

  previewRequest = previewRequestBuilder!!.build()
  captureSession!!.setRepeatingRequest(
    previewRequest!!,
    null, null
  )

修复内存泄漏。

此外，演示应用 Posenet.kt 被编码为使用 CPU，这可能有助于支持更大范围的设备，但将其切换到 GPU 或 NNAPI 会显着加快速度。