Pytorch 实现计算分类器准确率(总分类及子分类)


Posted in Python onJanuary 18, 2020

分类器平均准确率计算:

correct = torch.zeros(1).squeeze().cuda()
total = torch.zeros(1).squeeze().cuda()
for i, (images, labels) in enumerate(train_loader):
      images = Variable(images.cuda())
      labels = Variable(labels.cuda())

      output = model(images)

      prediction = torch.argmax(output, 1)
      correct += (prediction == labels).sum().float()
      total += len(labels)
acc_str = 'Accuracy: %f'%((correct/total).cpu().detach().data.numpy())

分类器各个子类准确率计算:

correct = list(0. for i in range(args.class_num))
total = list(0. for i in range(args.class_num))
for i, (images, labels) in enumerate(train_loader):
      images = Variable(images.cuda())
      labels = Variable(labels.cuda())

      output = model(images)

      prediction = torch.argmax(output, 1)
      res = prediction == labels
      for label_idx in range(len(labels)):
        label_single = label[label_idx]
        correct[label_single] += res[label_idx].item()
        total[label_single] += 1
 acc_str = 'Accuracy: %f'%(sum(correct)/sum(total))
 for acc_idx in range(len(train_class_correct)):
      try:
        acc = correct[acc_idx]/total[acc_idx]
      except:
        acc = 0
      finally:
        acc_str += '\tclassID:%d\tacc:%f\t'%(acc_idx+1, acc)

以上这篇Pytorch 实现计算分类器准确率(总分类及子分类)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python实现删除Android工程中的冗余字符串
Jan 19 Python
Python和GO语言实现的消息摘要算法示例
Mar 10 Python
Python抓取框架 Scrapy的架构
Aug 12 Python
Python3实现抓取javascript动态生成的html网页功能示例
Aug 22 Python
flask中使用蓝图将路由分开写在不同文件实例解析
Jan 19 Python
Python数据类型之Number数字操作实例详解
May 08 Python
Python 多个图同时在不同窗口显示的实现方法
Jul 07 Python
基于python解线性矩阵方程(numpy中的matrix类)
Oct 21 Python
Python打包模块wheel的使用方法与将python包发布到PyPI的方法详解
Feb 12 Python
浅析pip安装第三方库及pycharm中导入第三方库的问题
Mar 10 Python
python pandas利用fillna方法实现部分自动填充功能
Mar 16 Python
浅谈keras使用预训练模型vgg16分类,损失和准确度不变
Jul 02 Python
在pytorch 中计算精度、回归率、F1 score等指标的实例
Jan 18 #Python
Python中实现输入超时及如何通过变量获取变量名
Jan 18 #Python
Pytorch 计算误判率,计算准确率,计算召回率的例子
Jan 18 #Python
python:目标检测模型预测准确度计算方式(基于IoU)
Jan 18 #Python
Python实现计算长方形面积(带参数函数demo)
Jan 18 #Python
python实现的Iou与Giou代码
Jan 18 #Python
Python 简单计算要求形状面积的实例
Jan 18 #Python
You might like
smarty+adodb+部分自定义类的php开发模式
2006/12/31 PHP
php另类上传图片的方法(PHP用Socket上传图片)
2013/10/30 PHP
php函数指定默认值方法的小例子
2013/12/04 PHP
php 判断服务器操作系统的类型
2014/02/17 PHP
Yii列表定义与使用分页方法小结(3种方法)
2016/07/15 PHP
PHP生成图片验证码功能示例
2017/01/12 PHP
YII2框架中excel表格导出的方法详解
2017/07/21 PHP
浅谈Yii乐观锁的使用及原理
2017/07/25 PHP
始终在屏幕中间显示Div的代码(css+js)
2011/03/10 Javascript
js控制淡入淡出示例代码
2013/11/12 Javascript
jquery制作select列表双向选择示例代码
2014/09/02 Javascript
JavaScript中Array对象用法实例总结
2016/11/29 Javascript
js获取当前页的URL与window.location.href简单方法
2017/02/13 Javascript
angular仿支付宝密码框输入效果
2017/03/25 Javascript
react实现点击选中的li高亮的示例代码
2018/05/24 Javascript
Vue2.X 通过AJAX动态更新数据
2018/07/17 Javascript
Vue创建头部组件示例代码详解
2018/10/23 Javascript
微信小程序新闻网站详情页实例代码
2020/01/10 Javascript
vue解决跨域问题(推荐)
2020/11/10 Javascript
Python中用PIL库批量给图片加上序号的教程
2015/05/06 Python
基于python 字符编码的理解
2017/09/02 Python
Python提取特定时间段内数据的方法实例
2019/04/01 Python
Python对HTML转义字符进行反转义的实现方法
2019/04/28 Python
python3利用Socket实现通信的方法示例
2019/05/06 Python
python-Web-flask-视图内容和模板知识点西宁街
2019/08/23 Python
virtualenv介绍及简明教程
2020/06/23 Python
python 读txt文件,按‘,’分割每行数据操作
2020/07/05 Python
十八届三中全会报告学习材料
2014/02/17 职场文书
社区交通安全实施方案
2014/03/22 职场文书
应届生找工作求职信
2014/06/24 职场文书
2015年敬老院工作总结
2015/05/18 职场文书
2016年小学生新年寄语
2015/08/18 职场文书
2015年学校消防安全工作总结
2015/10/14 职场文书
快消品行业营销模式与盈利模式分享
2019/09/27 职场文书
高端收音机+蓝牙音箱,JBL TUNER FM带收音蓝牙音箱评测
2021/04/24 无线电
详解Vue的options
2021/05/15 Vue.js