FCN的分类的label是直接写类号码:1,2,3,...,255。
但softmaxwithLoss的第一个bottom,也就是网络输出,是要求向量。
即label=3的话,输出是0,0,0,1,0,0,...
所以写脚本注意一下,输出层的输出个数不能写1。
就是说softmaxwithLoss的两个bottom第二维数值是不一样的。
FCN的分类的label是直接写类号码:1,2,3,...,255。
但softmaxwithLoss的第一个bottom,也就是网络输出,是要求向量。
即label=3的话,输出是0,0,0,1,0,0,...
所以写脚本注意一下,输出层的输出个数不能写1。
就是说softmaxwithLoss的两个bottom第二维数值是不一样的。