【学习挑战赛 |深度学习实践（五）】识别眼睛状态

活动地址：CSDN21天学习挑战赛

学习的最大理由是想摆脱平庸，早一天就多一份人生的精彩；迟一天就多一天平庸的困扰。
热爱写作，愿意让自己成为更好的人…
…

  引言

	铭记于心
	✨我唯一知道的，便是我一无所知✨

【深度学习实践（五）】识别眼睛状态

1 卷积神经网络

CNN由纽约大学的Yann Lecun于1998年提出，其本质是一个多层感知机，成功的原因在于其所采用的局部连接和权值共享的方式，同时一种带有卷积结构的深度神经网络，卷积结构可以减少深层网络占用的内存量，其三个关键的操作，其一是局部感受野，其二是权值共享，其三是pooling层，有效的减少了网络的参数个数，缓解了模型的过拟合问题

• 一方面减少了权值的数量使得网络易于优化

• 另一方面降低了模型的复杂度，也就是减小了过拟合的风险

该优点在网络的输入是图像时表现的更为明显，使得图像可以直接作为网络的输入，避免了传统识别算法中复杂的特征提取和数据重建的过程，在二维图像的处理过程中有很大的优势，如网络能够自行抽取图像的特征包括颜色、纹理、形状及图像的拓扑结构，在处理二维图像的问题上，特别是识别位移、缩放及其他形式扭曲不变性的应用上具有良好的鲁棒性和运算效率等

2 模型训练

1 设置GPU

• GPU能够为大量数据的运算提供算力支持

import tensorflow as tf

gpus = tf.config.list_physical_devices("GPU")

if gpus:
    tf.config.experimental.set_memory_growth(gpus[0], True)  #设置GPU显存用量按需使用
    tf.config.set_visible_devices([gpus[0]],"GPU")

# 打印显卡信息，确认GPU可用
print(gpus)

2 加载训练集

• 读取数据集（训练集为验证码图片，都放在本地目录中，所以首先加载本地目录；为了检验是否读取成功，用matplotlib库显示图片），并将数据可视化

• 数据集官方介绍文档

  val_ds = tf.keras.preprocessing.image_dataset_from_directory(
      data_dir,
      validation_split=0.2,
      subset="validation",
      seed=12,
      image_size=(img_height, img_width),
      batch_size=batch_size)
  
  

  

3 数据预处理

• 将加载到内存的验证码图片训练集进行数据清洗，包括对无效图片，NULL值，图片大小统一的处理等等，下面封装，可以使过程更加简便，可以使过程更加简便

4 构建模型

下面就是比较重要的一步，构建模型

• 为了使训练的过程更加简便，所以可以使用prefetch函数：

  该函数的主要作用是 同时执行训练步骤的数据预处理阶段以及执行阶段，同时利用CPU与加速器，避免了传统训练过程中的“串行”思想（使用CPU进行数据预处理的时候加速器空闲，使用加速器进行模型训练的时候GPU空闲），缩短提取和转换数据所需的时间，使模型训练的效率大大提高

  

  • cache( )：将数据集以缓存的形式加载到内存中，以实现加速运行的目的

  • shuffle（）：将数据集打乱，增加随机性来使模型更具有普遍性

  AUTOTUNE = tf.data.AUTOTUNE
  
  train_ds = train_ds.cache().shuffle(1000).prefetch(buffer_size=AUTOTUNE)
  val_ds   = val_ds.cache().prefetch(buffer_size=AUTOTUNE)
  from tensorflow.keras import datasets, layers, models
  
  model = tf.keras.applications.VGG16()
  # 打印模型信息
  model.summary()
  
  
  

  

5 训练

将构建好模型后，就是将模型投入到实际使用环境中，基于神经网络对模型输入输出的反馈进行自身调整

• 设置初始学习率

  initial_learning_rate = 1e-4
  
  lr_schedule = tf.keras.optimizers.schedules.ExponentialDecay(
          initial_learning_rate, 
          decay_steps=20,      # 敲黑板！！！这里是指 steps，不是指epochs
          decay_rate=0.96,     # lr经过一次衰减就会变成 decay_rate*lr
          staircase=True)
  

• 将指数衰减学习率送入优化器

  optimizer = tf.keras.optimizers.Adam(learning_rate=lr_schedule)
  
  

• 编译与训练

  
  model.compile(optimizer=optimizer,
                loss     ='sparse_categorical_crossentropy',
                metrics  =['accuracy'])
  
  epochs = 10
  
  history = model.fit(
      train_ds,
      validation_data=val_ds,
      epochs=epochs
  )
  
  
  

• 得到训练数据变化趋势以及损失函数变化曲线，如下：

  

3 预测与评估

检验模型的训练效果，就需要使用模型对测试集进行预测与评估

采用加载的模型（new_model）来看预测结果

plt.figure(figsize=(10, 5))  # 图形的宽为10高为5
plt.suptitle("预测结果展示")

for images, labels in val_ds.take(1):
    for i in range(8):
        ax = plt.subplot(2, 4, i + 1)  
        
        # 显示图片
        plt.imshow(images[i].numpy().astype("uint8"))
        
        # 需要给图片增加一个维度
        img_array = tf.expand_dims(images[i], 0) 
        
        # 使用模型预测图片中的人物
        predictions = new_model.predict(img_array)
        plt.title(class_names[np.argmax(predictions)])

        plt.axis("off")

参考文章链接

写在最后：
路漫漫其修远兮，吾将上下而求索！伙伴们，再见！