【发布时间】:2021-05-05 03:59:54
【问题描述】:
我已经在 .h5 中训练了我的 Keras 模型。我的模型使用 6 个类,它能够使用图像对所有类进行分类。该模型能够输出它成功分类的类的名称。但是,我想在使用用户输入的图像测试模型时产生准确性。我已经到处搜索,但仍然没有解决这个问题。
model = load_model('prototype-tl2-80-20.h5')
classes = { 1:'Kacip Fatimah',
2:'Mempisang',
3:'Misai Adam',
4:'Pandan Serapat',
5:'Tapak Sulaiman',
6:'Tongkat Ali'}
image = Image.open(file_path)
image = image.resize((224,224))
image = numpy.expand_dims(image, axis=0)
image = numpy.array(image)
pred = model.predict_classes([image])[0]
sign = classes[pred+1]
print(sign)
【问题讨论】:
-
您所说的“生成准确性”究竟是什么意思?
-
我的意思是当用户将图像输入系统以对图像的类别进行分类时,系统可以对其进行分类,但有时会显示错误的分类。对于最终用户,他们不知道系统是否给出了正确的分类,因为没有关于它的信息。为了解决这个问题,我想输出每个图像的分类精度,但我仍然找不到解决方案。
-
啊,你问的是所谓的置信度或概率,它只是你模型的输出,使用model.predict并手动获取argmax,模型输出的最大值将给出你的概率,但在任何情况下,概率不会给你太多的信息。
-
非常感谢先生提供的信息。很高兴知道我对信心有错误的理解。
标签: python tensorflow keras