我正在尝试加载在TensorFlow中训练的mobilenet_v2_1.4_224（链接：https ://github.com/tensorflow/models/tree/master/research/slim/nets/mobilenet），并在GoLang中提供它。
问题是当我尝试用熊猫预测图像时，我得到了错误的预测。我的猜测是我下载了错误的 imagenet 标签文件。我上面提到的网络是在哪个 imagenet 数据集上训练的？
问问题
957 次
1 回答
2
来自 Tensorflow 模型动物园的 AFAIK MobileNet 在 ILSVRC-2012-CLS 上进行了训练，其中有 1000 个类，包括大熊猫和小熊猫。
但是，由于 MobileNet 并非 100% 准确，您可能会遇到不同的结果，具体取决于您提供的输入图像。确保将其紧紧围绕 Panda 进行裁剪，并尝试各种图像，因为即使是最好的 MobileNets V2 模型在 ImageNet 数据集上的准确率也约为 70%，有些可能会返回不正确的标签。
如果您希望能够更精确地对熊猫进行分类，我建议您使用熊猫数据集重新训练您的 MobileNet 模型。有很多关于如何做到这一点的可用示例，我在下面链接了几个使用 MobileNets 架构和 Tensorflow 作为框架的示例。
于 2019-03-05T12:01:47.077 回答