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

 在进行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)

 

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/292369.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Arduino平台软硬件原理及使用——Arduino Uno板卡的使用

本文目录 一、Arduino平台简介 二、Arduino Uno板卡简介 三、Arduino编译器简介 四、Arduino平台结合Arduino Uno板卡的使用 一、Arduino平台简介 Arduino 是一个灵活、易于使用的开源电子平台,这个平台包含软件和硬件。 Arduino能通过各种各样的传感器来感知环境&…

代码随想录刷题题Day19

刷题的第十九天,希望自己能够不断坚持下去,迎来蜕变。😀😀😀 刷题语言:C Day19 任务 ● 669. 修剪二叉搜索树 ● 108.将有序数组转换为二叉搜索树 ● 538.把二叉搜索树转换为累加树 1 修剪二叉搜索树 669…

数据库编程大赛:一条SQL计算扑克牌24点

你是否在寻找一个平台,能让你展示你的SQL技能,与同行们一较高下?你是否渴望在实战中提升你的SQL水平,开阔你的技术视野?如果你对这些都感兴趣,那么本次由NineData主办的《数据库编程大赛》,将是…

Leetcode 406 根据身高重建队列

题意理解: people [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]] 给定一个二维数组,(h,k)h表示此人身高,k表示前面有几个人比他高。 我们按照每个人的h,k两个维度的需求给每个人排在合适的位置。 如: [5,0][7,0]…

docker数据卷数据卷容器

前言 今天调休在家,随便玩玩,简单做下学习记录 1. 数据卷特点 数据卷在容器启动时初始化,如果容器使用的镜像在挂载点包含了数据,这些数据会被拷贝到新初始化的数据卷中数据卷可以在容器之间共享和重用可以对数据卷里的内容直接…

3. BlazorSignalRApp 结合使用 ASP.NET Core SignalR 和 Blazor

参考:https://learn.microsoft.com/zh-cn/aspnet/core/blazor/tutorials/signalr-blazor?viewaspnetcore-8.0&tabsvisual-studio 1.创建新项目 BlazorSignalRApp 2.添加项目依赖项 依赖项:Microsoft.AspNetCore.SignalR.Client 方式1 管理解决方案…

利用老毛桃、ultraiso软碟通制作启动U盘装系统 以及硬盘安装系统

目录 一. 老毛桃制作winPE镜像 1.1 准备工作 1.2 启动U盘制作步骤 1.3 启动U盘装系统 二. 使用ultraiso软碟通制作启动U盘 2.1 启动U盘制作步骤 2.2 启动U盘装系统 三. 硬盘安装系统 3.1 硬盘镜像制作步骤 3.2 硬盘镜像装系统 思维导图 一. 老毛桃制作winPE镜像 …

网络编程:信号、定时器、Libevent

1. 信号 (1)信号:由用户、系统或进程发送给目标进程的信息,以通知目标进程某个状态的改变或系统异常; 可由下述条件产生: 对前台进程,用户可以通过终端给它发送信号,如输入 CtrlC…

c++ qt 模态框和阻拦器 优先级 问题 修复 已解决

在c项目中。有 加载动画 和 模态框提醒的功能, 导致发生一个问题,有提示框的时候,动画也停止,必须点击 按钮 所有代码才能有效。 解决办法 谨慎使用 deleteLater,因为和模态框拦截有冲突, 使用 隐藏 或者 删除指针。 deleteLater 使用逻辑是 …

融资项目——vue之双向数据绑定

上一篇文章中使用的v-bind是单向绑定方法,即数据改变,网页相应的视图发生改变,但是网页视图发生改变其相关联的数据不会发生改变。但是双向数据绑定不同之处在于网页视图发生改变其相关联的数据也会发生改变。Vue可以使用v-model进行双向数据…

振动试验的工装夹具(GB/T 2423.43-2008)

但当试件体积较大,而且形状复杂时,这种固定方法显然很困难,这时需要制作夹具,让试件安装在夹具上然后把夹具牢固地固定在振动台面上,因此实际上夹具是试件与振动台面连接的过渡体,其功能是将振动台的振动和…

实在智能成功完成近2亿元C轮融资,全面迎接2024年Agent智能体应用元年

在这个最冷的季节,杭州实在智能科技有限公司(以下简称“实在智能”)依然表现火爆,近日,实在智能成功完成C轮融资近2亿元人民币,由金泰富资本和安吉智慧谷共同领投、安吉两山国创跟投。 在此轮融资以前&…