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开始标记。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值