0
点赞
收藏
分享

微信扫一扫

tensorflow将label数据转换为onhot编码(tf.one_hot的使用)


我们在进行验证码识别的时候,对于一组验证码的标签数据:

[nzpp,
fjzq
....]

对于每个字符转换为对应的数字(0-25):

[[13,25,15,15],[19,23,20,16]]

在全连接层进行softmax损失计算时需要根据标签的one_hot编码进,这时候我们就去需要转换

tf.one_hot(ary,depth,axis,on_value)

​ary​​​为要转换的数组
​​​depth​​​为要转换的类型个数,这里是26
​​​axis​​​为要转换的数据的维数,这里13,25,在数组的第二维所以为2
(0维[],1维[13,25,15,15],2维 13)
​​​on_value​​​ 为转换后唯一的值 这里为1.0
转换:

tf.one_hot([[13,25,15,15],[19,23,20,16]],depth=26,axis=2,on_value=1.0).eval()

tensorflow将label数据转换为onhot编码(tf.one_hot的使用)_字符转换


举报

相关推荐

0 条评论