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中的对象拷贝示例 python引用传递
Jan 23 Python
python模块之StringIO使用示例
Apr 08 Python
Python使用chardet判断字符编码
May 09 Python
简单掌握Python中glob模块查找文件路径的用法
Jul 05 Python
使用Python对Csv文件操作实例代码
May 12 Python
关于Django显示时间你应该知道的一些问题
Dec 25 Python
Python3读取Excel数据存入MySQL的方法
May 04 Python
python绘制立方体的方法
Jul 02 Python
Python PIL读取的图像发生自动旋转的实现方法
Jul 05 Python
python 并发编程 阻塞IO模型原理解析
Aug 20 Python
基于Python数据分析之pandas统计分析
Mar 03 Python
Python Mock模块原理及使用方法详解
Jul 07 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
php查询mysql数据库并将结果保存到数组的方法
2015/03/18 PHP
PHP中子类重载父类的方法【parent::方法名】
2016/05/06 PHP
PHP入门教程之字符串处理技巧总结(转换,过滤,解析,查找,截取,替换等)
2016/09/11 PHP
Yii2.0中使用js异步删除示例
2017/03/10 PHP
JavaScript 闭包深入理解(closure)
2009/05/27 Javascript
javascript 实用的文字链提示框效果
2010/06/30 Javascript
33个优秀的 jQuery 图片展示插件分享
2012/03/14 Javascript
js 表单提交后按钮变灰的实例代码
2013/08/16 Javascript
javascript中加号(+)操作符的一些神奇作用
2014/06/06 Javascript
JS实时弹出新消息提示框并有提示音响起的实现代码
2016/04/20 Javascript
几种二级联动案例(jQuery\Array\Ajax php)
2016/08/13 Javascript
jQuery中值得注意的trigger方法浅析
2016/12/12 Javascript
详解Vue 动态添加模板的几种方法
2017/04/25 Javascript
JS获取一个表单字段中多条数据并转化为json格式
2017/10/17 Javascript
jQuery实现表单动态添加数据并提交的方法
2018/07/19 jQuery
JavaScript中set与get方法用法示例
2018/08/15 Javascript
JavaScript继承与聚合实例详解
2019/01/22 Javascript
vue 实现input表单元素的disabled示例
2019/10/28 Javascript
详解React中共享组件逻辑的三种方式
2021/02/02 Javascript
Vue SPA 首屏优化方案
2021/02/26 Vue.js
[03:18]DOTA2亚洲邀请赛小组赛第一日 RECAP赛事回顾
2015/01/30 DOTA
详解C++编程中一元运算符的重载
2016/01/19 Python
python编程使用协程并发的优缺点
2018/09/20 Python
Python高斯消除矩阵
2019/01/02 Python
Python日期时间Time模块实例详解
2019/04/15 Python
将tensorflow模型打包成PB文件及PB文件读取方式
2020/01/23 Python
利用python生成照片墙的示例代码
2020/04/09 Python
Python脚本破解压缩文件口令实例教程(zipfile)
2020/06/14 Python
美国当红的名品折扣网:Gilt Groupe
2016/08/15 全球购物
简短大学毕业感言
2014/01/18 职场文书
《花木兰》教学反思
2014/04/09 职场文书
党员活动日总结
2014/05/05 职场文书
义诊活动总结
2015/02/04 职场文书
门卫管理制度范本
2015/08/05 职场文书
SQL实现LeetCode(178.分数排行)
2021/08/04 MySQL
spring IOC容器的Bean管理XML自动装配过程
2022/05/30 Java/Android