代码之家  ›  专栏  ›  技术社区  ›  rammanoj

基于树莓pi的图像分割

  •  1
  • rammanoj  · 技术社区  · 5 年前

    我一直在尝试用树莓皮进行图像分割。我搜索了不同的预训练模型来执行它,发现tensorflow lite,它有一个deeplab模型,它的尺寸非常小(2.7 Mb),可以用于物联网设备。但在我的例子中,我有一个自定义数据集,我需要在我的数据集上训练模型(即使用自定义数据集训练deeplab)。我的问题是raspberry Pi的内存和存储空间(相对而言)更少。所以,如果我用自定义数据集训练deeplab,我可以在raspberry Pi上运行它吗。如果是的话,有关于它的教程或研究论文吗?

    0 回复  |  直到 5 年前
        1
  •  0
  •   Kampi    5 年前

    你可以用 this 训练脚本。克隆 repository 然后跑 model.py 从…起 model/research/deeplab .

    我不会用覆盆子皮训练模型,因为它太慢了。更好的方法是在PC上训练它(可能有GPU支持),然后将模型导出到Raspberry Pi。