我想将图像数据集导入到带有图像和标签的 Numpy 数组中。我尝试使用 image_dataset_loader
来完成这个任务,目前已经编写了以下代码:
我的数据目录结构如下:
data -train -male -male_1.jpg -male_2.jpg -male_3.jpg -male_4.jpg -...... -female -female_1.jpg -female_2.jpg -female_3.jpg -female_4.jpg -...... -test -male -male_1.jpg -male_2.jpg -male_3.jpg -male_4.jpg -...... -female -female_1.jpg -female_2.jpg -female_3.jpg -female_4.jpg -......
我已经将所有图像调整为120×120的尺寸,并按照上述方式命名。每类别大约有56000个文件。当我运行上述脚本时,抛出了以下错误:
Traceback (most recent call last): File "main.py", line 33, in <module> (x_train, y_train), (x_test, y_test) = image_dataset_loader.load('./data', ['train', 'test']) File "/home/user/anaconda3/envs/AIOS/lib/python3.8/site-packages/image_dataset_loader.py", line 44, in load raise RuntimeError('Instance shapes are not consistent.')RuntimeError: Instance shapes are not consistent.
请问有人可以帮助我将这些图像整理到 Numpy 数组中吗?
回答:
检查你的图像的颜色。可能有些图像是灰度的。