首先路径如下图:
trai
首先导入plt和os
import os
import matplotlib.pyplot as plt
找到训练图片文件夹所在路径
train_dir = os.path.join("F:/暑假机器学习内容下载/暑假算法代码下载_杂/Unet/unet-master/data/membrane/train/image")
label_dir = os.path.join("F:/暑假机器学习内容下载/暑假算法代码下载_杂/Unet/unet-master/data/membrane/train/label")
print(train_dir)
结果:
Using TensorFlow backend.
F:/暑假机器学习内容下载/暑假算法代码下载_杂/Unet/unet-master/data/membrane/train/image
train_image_names = os.listdir(train_dir)
train_labels_names = os.listdir(label_dir)
print(train_image_names)
结果:
[‘0.png’, ‘1.png’, ‘10.png’, ‘11.png’, ‘12.png’, ‘13.png’, ‘14.png’, ‘15.png’, ‘16.png’, ‘17.png’, ‘18.png’, ‘19.png’, ‘2.png’, ‘20.png’, ‘21.png’, ‘22.png’, ‘23.png’, ‘24.png’, ‘25.png’, ‘26.png’, ‘