Raspberry Pi 4 (8 GB) 和 YOLOV4/YOLOV4-TINY 使用 Tensorflow-lite？

Posted 2023-03-16

【中文标题】Raspberry Pi 4 (8 GB) 和 YOLOV4/YOLOV4-TINY 使用 Tensorflow-lite？

【英文标题】：Raspberry Pi 4 (8 GB) with YOLOV4/YOLOV4-TINY using Tensorflow-lite?

【发布时间】：2021-12-24 08:41:40

【问题描述】：

使用 Tensorflow-lite 在 RPI 4 上运行 YOLOV4/YOLOV4-TINY 进行对象检测的最佳方法是什么？我想检测/计算编号。房间里有多少人使用它，然后检测椅子、香蕉等物品？

据我所知，这些库拥有 MIT 许可证，可用于教育/商业目的，对吗？

另外，在带有 tensorflow lite 的 Rpi 4 上，什么效果更好，是 YOLOv4 还是 YOLOv4 Tiny 还是其他什么？

谢谢。

【参考方案1】：

我自己没试过，但是 YOLOV4-TINY 的权重大小在 16-24MB 左右，和 MobileNet float 差不多。我认为这可能更适合像 RPi4 这样的小型设备。你能试一试，让我们知道它是否有效吗？ :)

据我所知，这些库拥有 MIT 许可证，可用于教育/商业目的，对吗？

TensorFlow 和 TensorFlow Lite 受 Apache 许可。您当然可以将其用于教育和商业目的。

【讨论】：

在这种情况下我需要探索 MobileNet，我会告诉你的。

【参考方案2】：

您可以查看 TFLite 对象检测 Raspberry Pi 示例。它使用在 Raspberry Pi 4 上运行良好的 EfficientDet-Lite 模型。 https://github.com/tensorflow/examples/tree/master/lite/examples/object_detection/raspberry_pi

如果您需要训练自定义模型，可以使用 Model Maker。请参阅此笔记本以了解更多信息。 https://github.com/khanhlvg/tflite_raspberry_pi/blob/main/object_detection/Train_custom_model_tutorial.ipynb