带张量流的图像检测器

Question

我想为图像上的自定义二进制形状构建一个简单的图像检测器。 我可能会在 ssd_inception_v2 等对象检测动物园上训练和使用模型。但它的效率非常低，因为它有数百兆字节的大小。 我什至无法想象在我的简单应用程序中使用它。谁能建议我如何解决这个问题？

我已经为我的图像构建了优秀的小型分类器。但无法构建小规模高效检测器。 （他们与检测框的位置）

Answer 1

我觉得你需要的是迁移学习。我会采用一种轻量级模型，例如 MobileNetV2 并在我的数据集上重新训练。 quick.If 如果您想进一步减小模型大小，请随意只采用 CNN 的前几层并重新训练它。由于您需要重新编写要使用的网络部分并使用预先训练的权重加载它，因此工作量会增加一些。