【发布时间】:2020-09-18 09:05:10
【问题描述】:
我有一个包含 200 张图像的列表(display_train_set)。我使用下面的代码制作的。我现在需要创建一个包含 40 个列表的列表,其中每个列表有 5 个来自 display_train_images 的图像。请帮助我是python新手。
dataset_dir = "FaceDataset"
for root, dirs, files in os.walk(dataset_dir):
path = root.split(os.sep)
img_dir = root + "/"
i = 0
for file in files:
img = imread(img_dir + file)
img_resized = resize(img, (200, 200))
if i <= 4:
display_test_set.append(img)
else:
display_train_set.append(img)
i += 1
【问题讨论】:
标签: python-3.x image-processing scikit-image os.walk