我尝试构建像这样的简单图像分类器 https://www.tensorflow.org/tutorials/images/classification
。
与此同时，我想在（特定的单独测试图像文件夹）的随机图像上打印图像标签
就像这张照片https://www.tensorflow.org/tutorials/keras/classification_files/output_oZTImqg_CaW1_0.png
和此代码（我的问题）是class_names和train_labels
plt.figure(figsize=(10,10))
for i in range(25):
plt.subplot(5,5,i+1)
plt.xticks([])
plt.yticks([])
plt.grid(False)
plt.imshow(train_images[i], cmap=plt.cm.binary)
plt.xlabel(class_names[train_labels[i]])
plt.show()
在此页面中 https://www.tensorflow.org/tutorials/keras/classification#preprocess_the_data
我的代码是从第一个链接复制的，但是我更改了数据集
PATH = r"C:\Users\memo-omen\Documents\py proj\i8mx1\dataset"
train_dir = os.path.join(PATH, 'train')
validation_dir = os.path.join(PATH, 'test')
train_cats_dir = os.path.join(train_dir, 'cats')
train_dogs_dir = os.path.join(train_dir, 'dogs')
validation_cats_dir = os.path.join(validation_dir, 'cats_test')
validation_dogs_dir = os.path.join(validation_dir, 'dogs_test')
等正如我说的一样，但我使用本地数据集
。
。
。
文件夹结构
c：\ dataset文件夹（主）
-
数据集\火车（文件夹）
数据集\火车\狗（文件夹）
img1.jpg
img2.jpg
数据集\火车\猫（文件夹）
img1.jpg
img2.jpg
-
数据集\测试（文件夹）
数据集\ test \狗（文件夹）
img1_test.jpg
img2_test.jpg
数据集\ test \猫（文件夹）
img1_test.jpg
img2_test.jpg