今天在使用Tensorflow做二分类分割时出现点小问题, 具体就是标签被被解码之后像素值全部都为零,仔细看看发现时是标签像素设置的问题:
标签中像素值是0和255, 而tf.one_hot(indices, depth)在解码时则会以0, 1,2… 的方式编码, 最大的元素值不能超过(depth - 1),depth表示像素类别数,所以应该将255像素值替换为1就行了!
今天在使用Tensorflow做二分类分割时出现点小问题, 具体就是标签被被解码之后像素值全部都为零,仔细看看发现时是标签像素设置的问题:
标签中像素值是0和255, 而tf.one_hot(indices, depth)在解码时则会以0, 1,2… 的方式编码, 最大的元素值不能超过(depth - 1),depth表示像素类别数,所以应该将255像素值替换为1就行了!