python - 如何在 python 中保存来自 U-NET 分割的分割图像？

Question

我已经基于这个链接在 Python 中训练了一个 U-NET 架构。

训练后，我想将单独分割的图像（当前是多个图像）保存在目录中。如何在 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)

标签： pythoncomputer-visionimage-segmentationcv2