def image_generator(image_list, batch_size):
for i in range(0, len(image_list), batch_size):
yield image_list[i:i + batch_size]
# 使用示例
image_list = ['image1.jpg', 'image2.jpg', 'image3.jpg', 'image4.jpg', 'image5.jpg']
batch_size = 2
for batch in image_generator(image_list, batch_size):
print(batch)
['image1.jpg', 'image2.jpg']
['image3.jpg', 'image4.jpg']