第48天-TensorFlow训练模型输出到web检测

一、项目设计

1. 前端使用Flask web框架上传图片，通过模型分析输出结果

2. 后端使用TensorFlow通过data文件夹图片训练模型

二、TensorFlow创建模型方法

import os
import numpy as np
import tensorflow as tf
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense

# 数据路径
data_dir = "data"
img_size = (150, 150)
batch_size = 32  # 增加批次大小

# 数据增强
datagen = ImageDataGenerator(
    rescale=1.0 / 255,
    rotation_range=20,  # 随机旋转
    width_shift_range=0.2,  #