如何解决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 举报,一经查实,本站将立刻删除。