Keras/ImageAI 对每张图片进行相同的分类

问题描述

我使用本网站作为指南 - https://towardsdatascience.com/train-image-recognition-ai-with-5-lines-of-code-8ed0bdd8d9ba

训练代码：

from imageai.Classification.Custom import ClassificationModelTrainer

model_trainer = ClassificationModelTrainer()
model_trainer.setModelTypeAsresnet50()
model_trainer.setDataDirectory("dataset")
model_trainer.trainModel(num_objects=3,num_experiments=10,enhance_data=True,batch_size=8,show_network_summary=True)

分类代码：

from imageai.Classification.Custom import CustomImageClassification
import os

execution_path = os.getcwd()

prediction = CustomImageClassification()
prediction.setModelTypeAsresnet50()
prediction.setModelPath("model_ex-010_acc-0.948454.h5")
prediction.setJsonPath("model_class.json")
prediction.loadModel(num_objects=3)

predictions,probabilities = prediction.predictimage("lettuce.jpg",result_count=3)

for eachPrediction,eachProbability in zip(predictions,probabilities):
    print(eachPrediction," : ",eachProbability)

该模型已经以 94% 的准确率进行了训练，但是无论图像如何，该模型都以 100% 的概率对每个相同的东西进行分类。我已按照网站上的指南进行操作，但似乎无法使其正常工作。

我的数据集包含 3 个类别，每个类别有 100 个训练图像，每个类别有 25 个测试图像，我还使用更大的数据集进行了测试，但仍然得到相同的结果。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）