类别名称转换为一一对应的数字标签

原创 2018年04月16日 11:00:08

在某些时候,例如使用mask_RCNN进行目标检测时,对每一幅图上有多个标记样本,那么就需要将类别的名字转换为数字标签传入到分类器中,比如,用颜色来举例:

'blue'=1

'yellow'=2

'red'=3

我们在输入类别的时候,不能输入['blue','yellow','red'],因为在里面进行计算LOSS,反向传播等等,都是利用数字在进行计算,而且仅仅只有['blue','yellow','red']这一种方式,还有['red','yellow','blue']等各种排列方式,那么:如何将['blue','yellow','red']转换位对应关系的[1,2,3]呢?

我们知道在python的字典中,是存在一一对应的关系的,也就是key-value键对,我们可以利用字典的这个特性将类别名称(字符串)与标签(数字)进行绑定,进而完成该对应转换!

首先,我们需要定义一个函数str2num:

            def str2num(s):
                digits = {'blue': 1, 'yellow': 2, 'red': 3, 'white': 4, 'black': 5}
                return digits[s]

这就完成了标签与数字的绑定,然后使用python自带的map函数对['blue','yellow','red']进行处理:

ID_list = list(map(str2num, ['blue','yellow','red']))

这样就完成了转换:     

            ID_list=[1,2,3]


pandas或者sklearn中如何将字符形式的标签数字化

pandas中是非常容易实现的,转换成Categories对象即可。术语叫做因子和水平,水平一般都会自动转成数字储存。c = ['A','A','A','B','B','C','C','C','C']...
  • Datapad
  • Datapad
  • 2017-09-05 15:31:56
  • 741

C# - 阿拉伯数字金额转换为繁体数字金额

  • 2011年10月11日 13:21
  • 739KB
  • 下载

caffe训练时根据文件名产生类别标号文件c++代码

caffe训练时,图片分为test集和train集,图片如下: 要产生文件列表 格式如下: test\300.jpg 3 test\301.jpg 3 test\400.jpg 4 t...
  • wuzuyu365
  • wuzuyu365
  • 2016-08-09 10:34:03
  • 322

python数据预处理之将类别数据转换为数值的方法

在进行python数据分析的时候,首先要进行数据预处理。 有时候不得不处理一些非数值类别的数据,嗯, 今天要说的就是面对这些数据该如何处理。 目前了解到的大概有三种方法: 1,通过L...
  • a33445621
  • a33445621
  • 2017-12-17 20:32:25
  • 559

VB数字小写转大写的源代码

  • 2009年09月21日 09:10
  • 6KB
  • 下载

PyTorch——Tensor_把索引标签转换成one-hot标签表示

对于分类问题,标签可以是类别索引值也可以是one-hot表示。以10类别分类为例,lable=[3] 和label=[0, 0, 0, 1, 0, 0, 0, 0, 0, 0]是一致的.现在给定索引标...
  • VictoriaW
  • VictoriaW
  • 2017-06-06 09:14:54
  • 6579

一道面试题,把excel的列的名称转换为数字

这是一道某公司的上机面试题: microsoft excel 中栏目是用 A,B,C...Z;AA,AB,...ZZ ... ZZZZ ....这样表示序列的它对应是这样 0,1,2,...25;26...
  • justin_bkdrong
  • justin_bkdrong
  • 2016-11-15 21:11:18
  • 534

将数字金额转换为中文大写金额

  • 2009年03月27日 09:38
  • 245KB
  • 下载

pandas将类别属性转化为数值属性的方法

原文地址 离散特征的编码分为两种情况: 1、离散特征的取值之间没有大小的意义,比如color:[red,blue],那么就使用one-hot编码 2、离散特征的取值有大小的意义,比如size:[...
  • chenpe32cp
  • chenpe32cp
  • 2017-07-19 22:48:00
  • 2694

TensorFlow 多分类标签转换成One-hot

TensorFlow 多分类标签转换成One-hot在处理多分类问题时,将多分类标签转成One-hot编码是一种很常见的手段,以下即为Tensorflow将标签转成One-hot的tensor。以Mn...
  • ictcxq
  • ictcxq
  • 2017-11-15 21:29:16
  • 1243
收藏助手
不良信息举报
您举报文章:类别名称转换为一一对应的数字标签
举报原因:
原因补充:

(最多只允许输入30个字)