用树莓派进行图像分割

Question

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

Answer 1

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

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