数据集的文字标签(label)转成数字标签

简介: 数据集的文字标签(label)转成数字标签

但愿有人能明白我标题的意思,因为自己在找相关解决办法的时候这样搜没有找到,所以写这篇博客记录下。

问题:

我要放入模型中的训练集和标签是这样的(都是numpy格式的文件):

1ecd1b2606ed46e9956a89f231c9802c.png

2020062310470442.png

标签集中共有1926个标签,10个类别,(类别前边的数字,代表的是对应的第几张图片)。

然后,直接把他放进模型中训练,但是把标签数组转换成张量的过程中出现了类型不匹配问题

Ytr = np.load('train1/'+'Ytr01.npy',allow_pickle=True)
y_data = torch.from_numpy(Ytr).reshape(1926,).type(torch.FloatTensor)
Traceback (most recent call last):
  File "train.py", line 81, in <module>
    y_data = torch.from_numpy(Ytr).reshape(1926,).type(torch.FloatTensor) #
TypeError: can't convert np.ndarray of type numpy.object_. The only supported types are: float64, float32, float16, int64, int32, int16, int8, uint8, and bool.

那么,接下来就是解决TypeError: can't convert np.ndarray of type numpy.object_.


思路就是把原标签集中标签的类型改为上述错误中的任一基本类型,可以直接将标签集中的每个类别转换成一个整数。


这里要用到sklearn.preprocessing.LabelEncoder这个工具,文档写了多种转换方式,我根据自己的需要选择了其中一种:

import numpy as np
from sklearn.preprocessing import LabelEncoder
#先把原标签集加载进来
y = np.load('Desktop/UCF10/train1/'+'Ytr01.npy')
#转换
le = preprocessing.LabelEncoder()
y=le.fit_transform(y)
#保存新标签集
np.save("Desktop/UCF10/train1/Ytr01", y)

就是这点代码!顺便把测试标签也转了

看下效果:

1ecd1b2606ed46e9956a89f231c9802c.png

自动转好~~,具体解释看官放文档,用哪个理解哪个。官方

再放进模型中就可以正常跑了。。。

相关文章
|
17小时前
|
前端开发
文本转换
文本转换。
8 2
|
17小时前
|
存储 编解码 前端开发
Base64编码与打印标签(label)实例
在本篇文章,我们认识什么是Base64编码格式,以及它内部的编码规则;也列举了base64的常用使用场景以及注意方面,同时也列举了实际场景使用与代码示例。最后列举几种常用base64转码为arrayBuffer,blod等格式。
56 0
AcGiGeometry::text 自画实体输出文字实例
AcGiGeometry::text 自画实体输出文字实例
|
17小时前
ArcGIS修改图层标签标注(Label)对应字段与内容的方法
ArcGIS修改图层标签标注(Label)对应字段与内容的方法
|
17小时前
uniapp里textarea多行文本输入限制数量
uniapp里textarea多行文本输入限制数量
|
17小时前
|
C++
[Qt5&控件] Label控件显示文本内容(字符串和整数)
[Qt5&控件] Label控件显示文本内容(字符串和整数)
73 0
[Qt5&控件] Label控件显示文本内容(字符串和整数)
textarea文本域控制字数多少(带数字,数字减小)demo效果示例(整理)
textarea文本域控制字数多少(带数字,数字减小)demo效果示例(整理)
|
10月前
Echarts参数属性学习:x轴标签文本过长自动缩减并替换成缩略号...
Echarts参数属性学习:x轴标签文本过长自动缩减并替换成缩略号...
80 0
|
11月前
|
JavaScript 前端开发
绚烂的文本换行,打破常规元素背景填充规则
在我们的日常开发中,为一个元素设置背景、边框、阴影、内外边距等,通常是会应用在整个元素上; 例如我为一个元素设置一个背景图,根据背景图的填充规则设定,这个图片可能会出现重复,或者拉伸,或者多余的地方留
78 0
绚烂的文本换行,打破常规元素背景填充规则
|
12月前
|
数据采集 算法 JavaScript
GNE 预处理技术——如何移除特定标签但是保留文字到父标签
GNE 预处理技术——如何移除特定标签但是保留文字到父标签
101 0