DL之AlDL之AlexNet:利用卷积神经网络类AlexNet实现猫狗分类识别(图片数据增强→保存h5模型)(二)

简介: DL之AlexNet:利用卷积神经网络类AlexNet实现猫狗分类识别(图片数据增强→保存h5模型)

基于ImageDataGenerator实现数据增强


扩充数据集大小,增强模型的泛化能力。比如进行旋转、变形、归一化等。


扩充数据量:对图像作简单的预处理(如缩放,改变像素值范围);

随机打乱图像顺序,并且在图像集上无限循环(不会出现数据用完的情况);

对图像加入扰动,大大增大数据量,避免多次输入相同的训练图像产生过拟合。

优化训练效率:训练神经网络时经常需要将数据分成小的批次(例如每16张图像作为一个batch提供给神经网络),在ImageDataGenerator中,只需要简单提供一个参数 batch_size = 16。



类AlexNet代码

n_channels = 3

input_shape = (*image_size, n_channels)

input_layer = Input(input_shape)

z = input_layer

z = Conv2D(64, (3, 3))(z)

z = BatchNormalization()(z)

z = Activation('relu')(z)

z = MaxPooling2D(pool_size=(2, 2))(z)

z = Conv2D(64, (3, 3))(z)

z = BatchNormalization()(z)

z = Activation('relu')(z)

z = MaxPooling2D(pool_size=(2, 2))(z)

z = Conv2D(128, (3, 3))(z)

z = BatchNormalization()(z)

z = Activation('relu')(z)

z = MaxPooling2D(pool_size=(2, 2))(z)

z = Conv2D(128, (3, 3))(z)

z = BatchNormalization()(z)

z = Activation('relu')(z)

z = MaxPooling2D(pool_size=(2, 2))(z)

z = Flatten()(z) # 将特征变成一维向量

z = Dense(64)(z)

z = BatchNormalization()(z)

z = Activation('relu')(z)

z = Dropout(0.5)(z)

z = Dense(1)(z)

z = Activation('sigmoid')(z)



相关文章
|
2月前
|
机器学习/深度学习 数据采集 人工智能
深度学习实战指南:从神经网络基础到模型优化的完整攻略
🌟 蒋星熠Jaxonic,AI探索者。深耕深度学习,从神经网络到Transformer,用代码践行智能革命。分享实战经验,助你构建CV、NLP模型,共赴二进制星辰大海。
|
3月前
|
机器学习/深度学习 传感器 算法
【无人车路径跟踪】基于神经网络的数据驱动迭代学习控制(ILC)算法,用于具有未知模型和重复任务的非线性单输入单输出(SISO)离散时间系统的无人车的路径跟踪(Matlab代码实现)
【无人车路径跟踪】基于神经网络的数据驱动迭代学习控制(ILC)算法,用于具有未知模型和重复任务的非线性单输入单输出(SISO)离散时间系统的无人车的路径跟踪(Matlab代码实现)
265 2
|
3月前
|
机器学习/深度学习 并行计算 算法
【CPOBP-NSWOA】基于豪冠猪优化BP神经网络模型的多目标鲸鱼寻优算法研究(Matlab代码实现)
【CPOBP-NSWOA】基于豪冠猪优化BP神经网络模型的多目标鲸鱼寻优算法研究(Matlab代码实现)
100 8
|
4月前
|
算法 安全 网络安全
【多智能体系统】遭受DoS攻击的网络物理多智能体系统的弹性模型预测控制MPC研究(Simulink仿真实现)
【多智能体系统】遭受DoS攻击的网络物理多智能体系统的弹性模型预测控制MPC研究(Simulink仿真实现)
239 0
|
3月前
|
机器学习/深度学习 算法 调度
14种智能算法优化BP神经网络(14种方法)实现数据预测分类研究(Matlab代码实现)
14种智能算法优化BP神经网络(14种方法)实现数据预测分类研究(Matlab代码实现)
407 0
|
2月前
|
机器学习/深度学习 数据采集 存储
概率神经网络的分类预测--基于PNN的变压器故障诊断(Matlab代码实现)
概率神经网络的分类预测--基于PNN的变压器故障诊断(Matlab代码实现)
350 0
|
3月前
|
机器学习/深度学习 缓存 算法
2025年华为杯A题|通用神经网络处理器下的核内调度问题研究生数学建模|思路、代码、论文|持续更新中....
2025年华为杯A题|通用神经网络处理器下的核内调度问题研究生数学建模|思路、代码、论文|持续更新中....
461 1
|
3月前
|
机器学习/深度学习 传感器 算法
【表面粗糙度】基于粒子群PSO算法优化-BP神经网络的表面粗糙度研究(Matlab代码实现)
【表面粗糙度】基于粒子群PSO算法优化-BP神经网络的表面粗糙度研究(Matlab代码实现)
246 7
|
3月前
|
机器学习/深度学习 算法 数据可视化
PINN物理信息神经网络用于求解二阶常微分方程(ODE)的边值问题研究(Matlab代码实现)
PINN物理信息神经网络用于求解二阶常微分方程(ODE)的边值问题研究(Matlab代码实现)
346 6
|
3月前
|
机器学习/深度学习 传感器 分布式计算
基于模糊RBF神经网络轨迹跟踪研究(Matlab代码实现)
基于模糊RBF神经网络轨迹跟踪研究(Matlab代码实现)
220 1

热门文章

最新文章