一、数据集加载:
纯手工打造一个函数用来加载数据,数据分别为image和mask
我们所需要的类型:(B数量,长,宽,通道数),只要将数据加载成这样并mask和image对应就行。
第一步:
获取文件中所有的图片文件,并进行排序(mask和label的标签最好对应起来,这样对文件名进行排序就可以自动对齐)
train_mask = sorted(glob.glob('/kaggle/input/image-xy/Train_Label/*.jpg'))
train_image = sorted(glob.glob('/kaggle/input/image-xy/Train_image/*.jpg'))
第二步:
读取