solution for all black mask （结果为全黑的解决方法）

[gaoming714]

No description provided.

[gaoming714]

file data.py
modify the saveResult function
I use cv2, because i am not familiar with io.image for output.

import cv2

def saveResult(save_path,npyfile,flag_multi_class = False,num_class = 2):
for i,item in enumerate(npyfile):
img = labelVisualize(num_class,COLOR_DICT,item) if flag_multi_class else item[:,:,0]
print(img) ## ex. [[0.16509348 0.08227982 ... 0.07113015]
img = (img > 0.5).astype(np.uint8) # .reshape(256, 256)
print(img) ## ex. [[ 0 0 0 0 ... 1 1 1 ]]
img = (img * 255)
print(img) ## ex. [[0 0 00 ... 255 255 255]]
# img = labelVisualize(num_class,COLOR_DICT,item) if flag_multi_class else item[:,:,0]
# io.imsave(os.path.join(save_path,"%d_predict.png"%i),img)
cv2.imwrite(os.path.join(save_path,"%d_predict.png"%i), img)

[Hoon-Park0618]

thx. thx. thx. thx.!!!!