树莓派图像分割

Posted 2023-03-27

【中文标题】树莓派图像分割

【英文标题】：Image segmentation with raspberry pi

【发布时间】：2020-03-23 16:16:30

【问题描述】：

我一直在尝试用树莓派进行图像分割。我搜索了不同的预训练模型来执行它并遇到了 tensorflow lite，它有一个 deeplab 模型，它的大小非常小（2.7 Mb），可用于 IOT 设备。但就我而言，我有一个自定义数据集，我需要在我的数据集上训练模型（即使用自定义数据集训练 deeplab）。我的问题是树莓派的 RAM 和存储空间较少（相对而言）。所以，如果我用自定义数据集训练 deeplab，我可以在树莓派上运行它吗？如果有，是否有相关的教程或研究论文？

【参考方案1】：

您可以使用this 训练脚本。克隆repository 并从model/research/deeplab 运行model.py。

我不会在 Raspberry Pi 上训练模型，因为它太慢了。更好的方法是在 PC 上训练它（可能支持 GPU）并将模型导出到 Raspberry Pi。

【讨论】：

它是否在树莓派中实时预测？

这取决于您输入的图像（大小、通道）、图形大小等，您必须尝试一下。也许您使用原始模型并在 Raspberry Pi 上进行测试。您可以添加某种加速器设备（即 Edge TPU USB-Stick）来加快分类速度。