在图片上画出mask和pred

画出论文中《Variance-aware attention U-Net for multi-organ segmentation》的图1，也就是在原图上画出mask和pred的位置。

新建一个文件夹

然后运行代码：

import cv2
import os
from os.path import splitext

####第一次：把GT（绿）放在原图上
###第二次：把pre（红）放在第一次输出的图片上
###需要保证预测输出的图片和原图大小一样(256*256)


if __name__== '__main__':
    # image = cv2.imread('./image/02_4.png')    ###在原图上画masks轮廓, 绿
    # mask = cv2.imread('./mas