PaddlePaddle图像预测部分预测的没有输出类别的名称

本文详细介绍了在使用PaddlePaddle预测模型时遇到的输出为整数而非类别名称的问题,并提供了具体的解决方案,即通过映射整数标签到相应的类别名称来解决。同时,文章还探讨了PaddlePaddle中分类标签的特性,帮助读者更好地理解和应用预测模型。
部署运行你感兴趣的模型镜像
  • 问题描述:在使用预测模型预测图片的时候,输出的是一个整数,而不是图片的类别名称。

  • 报错信息:

infer results:  5
  • 问题复现:使用训练保存的预测模型预测经过预处理的图片,使用预测的结果经过np.argmax(results[0])进行处理,获取概率最大的值,最后输出的是一个整数值。复现代码如下:
img = load_image('dog.png') 
results = inferencer.infer({'pixel': img})
print("infer results: ", np.argmax(results[0]))
  • 解决问题:在训练的时候,使用的是类别的标签,也就是一个整数,所以在预测的时候输出的也是类别的标签。需要输出类别的名称,还要根据标签对应的类别名称进行输出。
img = load_image('dog.png')
results = inferencer.infer({'pixel': img})

label_list = [
    "airplane", "automobile", "bird", "cat", "deer", "dog", "frog", "horse",
    "ship", "truck"]
print("infer results: ", label_list[np.argmax(results[0])])
  • 问题拓展:PaddlePaddle的分类标签都是整数值,其中int64类型的标签非常常用。PaddlePaddle的分类标签不支持使用字符串的方式输入类别名称,所以都是整数作为类别的标签,而已必须要从0开始标记。

您可能感兴趣的与本文相关的镜像

PaddlePaddle-v3.3

PaddlePaddle-v3.3

PaddlePaddle

PaddlePaddle是由百度自主研发的深度学习平台,自 2016 年开源以来已广泛应用于工业界。作为一个全面的深度学习生态系统,它提供了核心框架、模型库、开发工具包等完整解决方案。目前已服务超过 2185 万开发者,67 万企业,产生了 110 万个模型

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值