tf.keras.utils.image_dataset_from_directory 找不到图像

Question

我有简单的代码：

data_dir_str = "./photo/pults/samsung_small"
data_dir = pathlib.Path(data_dir_str)
image_count = len(list(data_dir.glob('*.*')))
print('images in directory: ' + str(image_count)) # 12 files in directory

sams_pults = list(data_dir.glob('*.jpg'))
img = PIL.Image.open(str(sams_pults[0]))
img_width, img_height = img.size
#img.show()  # This works well - shows image

batch_size = 10

train_ds = tf.keras.utils.image_dataset_from_directory(
  data_dir_str, #data_dir,
  validation_split=0.2,
  subset="training",
  #seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size
)

所有图像均为 jpg 格式。 使用img.show时我可以看到一张图片。 但是image_dataset_from_directory给了我错误：“在目录 {directory} 中找不到图像。允许的格式：{ALLOWLIST_FORMATS}”

我不明白出了什么问题。

Answer 1

我找到了解决方案：

./photo/pults/samsung_small -> ./photo/pults

路径不需要设置为放置图像的目录，而是上一层。 因为目录名是用来分类的。