具有张量流的图像检测器

问题描述 投票:1回答:1

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

我已经为我的图像建立了出色的小型分类器。但无法构建小型高效检测器。 (带有检测框的位置)

tensorflow keras object-detection object-detection-api
1个回答
0
投票

我认为您需要的是转移学习。我将采用MobileNetV2之类的轻量模型之一,并在我的数据集中进行重新训练。它应该很快,如果您想进一步减小模型尺寸,可以随意只使用CNN的前几层并对其进行重新训练。由于您需要重新编写要使用的网络部分并使用预先训练的权重进行加载,因此需要做更多的工作。

© www.soinside.com 2019 - 2024. All rights reserved.