我们在进行验证码识别的时候,对于一组验证码的标签数据:
[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()