【问题标题】:model.predict_classes vs model.predict_generator in keraskeras 中的 model.predict_classes 与 model.predict_generator
【发布时间】:2017-05-03 04:37:40
【问题描述】:

我了解 predict_generator 输出概率。为了得到这个类,我就找到了最大概率的索引,这将是最可能的类。但是我发现这样做之后,我得到的输出与调用 predict_classes 不同。我不明白为什么。有人可以解释一下吗?

【问题讨论】:

    标签: machine-learning neural-network deep-learning keras conv-neural-network


    【解决方案1】:

    Keras 中的生成器使用 glob 列出按字母顺序排序的文件夹,您可以使用

    获取训练期间使用的类
    # save classes to JSON
    class_json = json.dumps(train_generator.class_indices)
    with open("class.json", "w") as class_file:
        class_file.write(class_json)
    

    在批次生成器(此处)中对样本进行混洗,以便在 fit_generator 或评估生成器请求批次时给出随机样本。

    如果对图像执行此操作,另一种可能性是不使用 https://github.com/fchollet/keras/issues/3477 中提到的 ImageDataGenerator 中的 rescale=1./255

    希望有所帮助!

    【讨论】:

      猜你喜欢
      • 2019-04-21
      • 2019-08-06
      • 2017-12-20
      • 1970-01-01
      • 1970-01-01
      • 2021-01-31
      • 2017-03-26
      • 1970-01-01
      • 2020-12-20
      相关资源
      最近更新 更多