【问题标题】:Loaded MobileNet model giving wrong predictions加载的 MobileNet 模型给出错误的预测
【发布时间】:2019-03-05 20:01:47
【问题描述】:

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

当我尝试用熊猫预测图像时,我得到了错误的预测。我的猜测是我下载了错误的 imagenet 标签文件。我上面提到的网络是在哪个 imagenet 数据集上训练的?

【问题讨论】:

    标签: tensorflow go imagenet


    【解决方案1】:

    Tensorflow Model Zoo 的 AFAIK MobileNet 在 ILSVRC-2012-CLS 上进行了训练,其中包含大熊猫和小熊猫等 1000 个类。

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

    如果您希望能够更精确地对熊猫进行分类,我建议您使用熊猫数据集重新训练您的 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

    【讨论】:

    猜你喜欢
    • 2018-06-12
    • 2020-10-17
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2019-05-24
    相关资源
    最近更新 更多