tensorflow - 加载的 MobileNet 模型给出了错误的预测

Question

我正在尝试加载在 TensorFlow 中训练的 mobilenet_v2_1.4_224(链接:https://github.com/tensorflow/models/tree/master/research/slim/nets/mobilenet)，并在 < strong>GoLang.

事情是当我尝试用 Pandas 对图像进行预测时，我得到了错误的预测。我的猜测是我下载了错误的 imagenet 标签文件。我上面提到的网络是在哪个 imagenet 数据集上训练的？

Answer 1

来自 Tensorflow Model Zoo 的 AFAIK MobileNet 是在 ILSVRC-2012-CLS 上训练的，它有 1,000 个类别，包括 Pandas 和小 Pandas 。

但是，由于 MobileNet 并非 100% 准确，您可能会遇到不同的结果，具体取决于您提供的输入图像。确保它在 Pandas 周围紧密裁剪，并尝试各种图像，因为有些图像可能会返回不正确的标签，因为即使是最好的 MobileNets V2 模型在 ImageNet 数据集上的准确率也约为 70%。

如果您希望能够更准确地对 Pandas 进行分类，我建议您使用 Pandas 数据集重新训练您的 MobileNet 模型。有很多关于如何执行此操作的可用示例，我在下面链接了几个使用 MobileNets 架构和 Tensorflow 作为框架的示例。

https://www.tensorflow.org/hub/tutorials/image_retraining

https://hackernoon.com/creating-insanely-fast-image-classifiers-with-mobilenet-in-tensorflow-f030ce0a2991

https://medium.com/@sumit.arora/training-a-neural-network-using-mobilenets-in-tensorflow-for-image-classification-on-android-14f2792f64c1