Python keras

有几张猫狗的图片对应标签是1,2,3等等keras数据集处理个人读取喜欢用PIL应怎么写

方案来自 梦想橡皮擦 狂飙组基于 GPT 编写的 “程秘”


from PIL import Image
import numpy as np
img = Image.open("dog_or_cat_image.jpg")


img_array = np.array(img)

根据标签将图像分类:
假设图像的标签存储在一个列表中,例如:

labels = [1, 2, 3, ...]

可以根据标签对图像进行分类,例如:

dog_images = []
cat_images = []

for i in range(len(img_array)):
    if labels[i] == 1:
        dog_images.append(img_array[i])
    elif labels[i] == 2:
        cat_images.append(img_array[i])
    # ...

然后,使用dog_images和cat_images数组进行模型训练和评估