栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

无法在Keras中使用VGG19预测单个图像的标签

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

无法在Keras中使用VGG19预测单个图像的标签

depre_predictions
用于根据ImageNet数据集中具有1000个类别的类别标签对模型的预测进行解码。但是,您经过微调的模型只有12个类。因此,
depre_predictions
在这里使用没有意义。当然,您必须知道这12个类别的标签是什么。因此,只需在预测中取最大分数的索引并找到其标签即可:

# create a list containing the class labelsclass_labels = ['class1', 'class2', 'class3', ...., 'class12']# find the index of the class with maximum scorepred = np.argmax(class_labels, axis=-1)# print the label of the class with maximum scoreprint(class_labels[pred[0]])


转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/429276.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号