前言

开源项目unet中使用的图像尺寸是固定的256，但是如果要使用resize图片尺寸的话，调用OpenCV中的resize函数不可行，因为源码加载图片的格式是ndarray，所以必须使用与格式对应的resize格式，修改代码如下：

        img = np.resize(img, (1, 640, 400, 1))
        mask = np.resize(img, (1, 640, 400, 1))