the target and pred are all range from 0 to 255.I use the channel-wise argmax op to generate the pred ndarray,and then times 255 cause the label is 0 or 1, so times 255 to make it show as an image in black and white.
QQ截图20180706200258.png1451×392 17.9 KB

that’s the result follow ur guidance.