使用拥抱脸和 ViT 无法加载灰度图像

Question

我有一个

grayscale

格式的图像数据集。现在我想使用视觉变压器和拥抱脸对其进行分类。 数据集位于谷歌驱动器中，但问题是我无法读取图像。可能的原因是什么？我使用以下代码来读取图像。

from datasets import load_dataset
dataset = load_dataset("ImageFolder", data_dir="/content/Xray8_New_DataSet")

使用 ViT 和 Hugging Face 从 google Drive 读取图像。

Answer 1

注意拼写。数据集构建器名称是小写的

imagefolder

:

from datasets import load_dataset
d = load_dataset("imagefolder", data_dir="/home/cronoik/downloads/bla/")