1

我正在尝试从 Tensorflow 2 中获取不同的 SSD 模型以在 Android 中工作。示例 .tflite 文件工作正常,但我转换并尝试过的不同其他模型似乎都有相同的问题。

我主要遵循本指南来了解如何将这些模型转换为移动设备:https ://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/running_on_mobile_tf2.md 这一切都很好,但是在尝试时要在移动设备上运行它,检查的第一帧会引发此错误:

初始化 ObjectDetector 时发生错误:索引 0 处的输出张量预计有 3 个维度,发现为 2。

我尝试过的模型:

  • SSD MobileNet V2 FPNLite 640x640
  • SSD ResNet152 V1 FPN 1024x1024 (RetinaNet152)
  • SSD ResNet101 V1 FPN 640x640 (RetinaNet101)

有谁知道我可能做错了什么?

编辑:我发现了几天前打开的类似问题。可能相关:https ://github.com/tensorflow/tensorflow/issues/51591#issuecomment-905216674

4

0 回答 0