pytorch和tensorflow实现one-hot编码,label编码为target
在我们实现网络训练或者自定义loss函数式,要计算预测图score和GT图target之间的loss时,score和target的shape需要保持一致,而标签数据label通常只是索引信息,我们需要将label数据通过onehot编码转化为向量(矩阵)target,在target中对应label索引的位置为1.0,其余为0.0,通常label类型为int,转化后的target为float首先...
原创
2019-08-01 22:08:42 ·
1997 阅读 ·
0 评论