Ошибка unet показывает, что "низкоконтрастное изображение io.imsave (os.path.join (save_path,"% d_predict.png "% i), img_as_ubyte (img))" - PullRequest
0 голосов
/ 25 февраля 2020

Я нахожу информацию о коде, которая https://github.com/zhixuhao/unet в Google. Уже обучение работает хорошо, но код ошибки выводится из теста. Что еще интереснее, это то, что оно было реализовано в тесте, но когда я его снова реализовал, я получил код ошибки. Я думаю, что "def testGenerator (test_path, num_image = 30, target_size = (256,256), flag_multi_class = True, as_gray = False):" в datd.py

, пожалуйста. почему эти коды ошибок там? "низкоконтрастное изображение io.imsave (os.path.join (save_path,"% d_predict.png "% i), img_as_ubyte (img))"

def testGenerator(test_path,num_image = 30,target_size = (256,256),flag_multi_class = True,as_gray = False):
for i in range(num_image):
    img = io.imread(os.path.join(test_path,"%d.png"%i),as_gray = as_gray)
    img = img / 255
    img = trans.resize(img,target_size)
    #img = np.reshape(img,img.shape+(1,)) if (not flag_multi_class) else img
    img = np.reshape(img,(1,)+img.shape)
    yield img

C:\Users\woong\Desktop\unet-master\unet-master\data.py:125: UserWarning: data/membrane/test\29_predict.png is a low contrast image io.imsave(os.path.join(save_path,"%d_predict.png"%i),img_as_ubyte(img))
...