有几张猫狗的图片对应标签是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数组进行模型训练和评估