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

在某些时候,例如使用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]


阅读更多
个人分类: 深度学习
上一篇could be decompression bomb DOS attack
下一篇从github下载开发者版本的源码进行编译安装scikit-image(0.14)
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭