树莓派图像分割

Question

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

Answer 1

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

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