pytorch-模型预测概率值为负数

2023-12-21 10:49:19

?在进行ocr识别模型预测的时候，发现预测的结果是正确的，但是概率值是负数：

net_out = net(img) #torch.Size([70, 1, 41]) 
logit, preds = net_out.max(2) #41是类别 需要对类别取最大值
preds = preds.transpose(1, 0).contiguous().view(-1) #torch.Size([70]) 70是序列，每个序列找到对应的类别了
lab2str,char_logit = decode_out(preds,logit, args.characters) #最终输出lab2str代表字符连起来的字符串，char_logit代表每个字符的概率值

于是添加了softmax之后，概率值恢复正常：

net_out = net(img)
######## 添加softmax ########
net_out = torch.softmax(net_out, dim=2)
logit, preds = net_out.max(2)
preds = preds.transpose(1, 0).contiguous().view(-1)
lab2str,char_logit = decode_out(preds,logit, args.characters)

?

文章来源:https://blog.csdn.net/weixin_40446557/article/details/135122602
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。