1. 安装必要的库
首先,确保你已经安装了必要的Python库。如果没有安装,请运行以下命令:
bash复制代码
pip install numpy matplotlib tensorflow opencv-python |
2. 图像预处理
我们将使用OpenCV来加载和预处理图像数据。假设你有一个图像数据集,每个类别的图像存放在单独的文件夹中。
python复制代码
import os |
|
import cv2 |
|
import numpy as np |
|
import matplotlib.pyplot as plt |
|
from tensorflow.keras.utils import to_categorical |
|
from sklearn.model_selection import train_test_split |
|
from tensorflow.keras.preprocessing.image import ImageDataGenerator |
|
# 定义数据集的路径 |
|
data_dir = 'path/to/your/dataset' |
|
image_size = (64, 64) # 调整为你需要的图像尺寸 |
|
batch_size = 32 |
|
# 使用ImageDataGenerator来加载和预处理数据 |
|
train_datagen = ImageDataGenerator(rescale |