一、报错截图:
二、分析原因:
ValueError可知,我们的数据格式可能有问题,这里需要shape一下我们的数据,看一下数据的大小是否一致。
# 将数据集打乱随机分组
x_train, x_test, y_train, y_test = train_test_split(imgs,
labels,
test_size=0.4,
random_state=random.randint(0, 100))
for i in x_train:
print(i.shape)
结果发现,数据中有些图像的大小不一致。
三、挑出数据中不一样的数据,并保存:
# Inspired by: DuckSoft <realducksoft@gmail.com>