Как сохранить сегментированные изображения из сегментации U-NET в Python? - PullRequest
Новая
       59

Как сохранить сегментированные изображения из сегментации U-NET в Python?

0 голосов
/ 13 мая 2019

Я обучил архитектуре U-NET на основе Python эту ссылку.

После обучения я хочу сохранить индивидуально сегментированные изображения (которые в настоящее время являются несколькими изображениями) в каталоге.Как сохранить эти сегментированные изображения в Python?Я использую код ниже, чтобы сохранить результаты.

def saveResults(a,b):        
    preds_test = a.predict(b, verbose=1)
    cutoff=0.5
    a = 'C:/Users/ultim\Desktop/Rest/testpredict'
    for i in range(len(preds_test)):
        lab_img = label(preds_test[i] > cutoff)
        for i in range(1, lab_img.max() + 1):
            lab_img[i] = np.array(lab_img[i], dtype=np.uint8)
            cv2.imwrite('a' + i, lab_img[i])
saveResults(modelZ,X_test)
...