首先路径如下图:
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’, ‘27.png’, ‘28.png’, ‘29.png’, ‘3.png’, ‘4.png’, ‘5.png’, ‘6.png’, ‘7.png’, ‘8.png’, ‘9.png’]
显示图片:
im = plt.imread(train_dir +"/" + train_image_names[0] ) #第0个图片
plt.imshow(im)
plt.show()
结果: