其他分享
首页 > 其他分享> > 【535】语义分割 label 通道与模型输出通道的

【535】语义分割 label 通道与模型输出通道的

作者:互联网

mirrors / fengshuanglang / unet 

  对于语音分割为两类的情况,label 的值为 0 和 1,预测结果也是只有 0 和 1,存在集中可行的情况:

  1. label 放在 1 个通道,模型输出可以是 1 个通道,包含 0 和 1,用 sigmoid 作为最后的激活函数
  2. label 放在 1 个通道,模型输出可以是 2 个通道,0 通道预测 0 类,1 通道预测 1 类,用 softmax 作为最后的激活函数
  3. label 放在 2 个通道,相当于将 0 和 1 的值做 one-hot 编码,模型输出是 2 个通道,0 通道预测 0 类,1 通道预测 1 类,用 softmax 作为最后的激活函数

  对比下来比较容易理解及操作的就是都是 1 个通道,对于自定义 loss 操作起来更加方便

标签:输出,预测,模型,label,535,softmax,通道
来源: https://www.cnblogs.com/alex-bn-lee/p/15110182.html