如何在python中保存UNET分割的分割图像？

Question

我已经用基于Python的this链接训练了一个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)