实战Keras3.0：自定义图片数据集分类任务

原创

已于 2024-01-03 13:08:11 修改 · 1.1k 阅读

15 ·

CC 4.0 BY-SA版权

文章标签：

#python #机器学习 #深度学习 #分类 #keras

于 2024-01-02 21:59:42 首次发布

一、创建自定义图片数据集

1、数据收集

以10张小狗图片和10张小猫图片为例

2、数据预处理

1、创建Excel表格，并在其中创建两列，一列是图片路径，另一列是对应的标签（狗0、猫1）

2、用pandas库的read_excel函数读取Excel，用PIL库的Image函数将图片数据格式化

import pandas as pd
from sklearn.model_selection import train_test_split
import keras
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

#读取Excel表格
data = pd.read_excel('/Users/Desktop/cat_dog/dog_and_cat.xlsx')

#图片数据处理
def imageDigitization(image_data):
    train_data = []
    for image_item  in image_data: