TensorFlow_T7 咖啡豆识别

一、前言

二、前期准备

1、设置GPU

2、导入数据

3、查看数据图片

三、数据预处理

1、加载数据

2、可视化数据

3、配置数据集

四、构建VGG-16网络

1、VGG优缺点分析

2、自建模型

3、网络结构图

五、编译

六、训练模型

一、前言

🍨 本文为🔗365天深度学习训练营中的学习记录博客
🍖 原作者：K同学啊

二、前期准备

1、设置GPU

import tensorflow as tfgpus=tf.config.list_physical_devices("GPU")if gpus:tf.config.experimental.set_memory_growth(gpus[0],True)tf.config.set_visible_devices([gpus[0]],"GPU")
gpus

2、导入数据

import pathlibdata_dir="D:\THE MNIST DATABASE\T7"
data_dir=pathlib.Path(data_dir)

3、查看数据图片

image_count=len(list(data_dir.glob('*/*.png')))print("图片总数为：",image_count)

运行结果：

三、数据预处理

1、加载数据

使用image_dataset_from_directory方法将磁盘中的数据加载到tf.data.Dataset中。

train_ds=tf.keras.preprocessing.image_dataset_from_directory(data_dir,validation_split=0.2,subset="training",seed=123,image_size=(224,224),batch_size=32
)

运行结果如下：

加载验证集：

val_ds=tf.keras.preprocessing.image_dataset_from_directory(data_dir,validation_split=0.2,subset="validation",seed=123,image_size=(224,224),batch_size=32
)

运行结果如下：

通过class_names输出数据集的标签。标签将按字母顺序对应于目录名称。

class_names=train_ds.class_names
print(class_names)

运行结果如下：

2、可视化数据

import matplotlib.pyplot as pltplt.figure(figsize=(10,4))for images,labels in train_ds.take(1):for i in range(10):ax=plt.subplot(2,5,i+1)plt.imshow(images[i].numpy().astype("uint8"))plt.title(class_names[labels[i]])plt.axis("off")

运行结果如下：

查看图像格式：

for image_batch,labels_batch in train_ds:print(image_batch.shape)print(labels_batch.shape)break

运行结果如下：

3、配置数据集

shuffle() ：打乱数据；
prefetch() ：预取数据，加速运行；
cache() ：将数据集缓存到内存当中，加速运行；

对数据集进行预处理，对于验证集只进行了缓存和预取操作，没有进行打乱操作。

AUTOTUNE=tf.data.AUTOTUNEtrain_ds=train_ds.cache().shuffle(1000).prefetch(buffer_size=AUTOTUNE)
val_ds=val_ds.cache().prefetch(buffer_size=AUTOTUNE)

对图像数据集进行归一化。将输入数据除以255，将像素值缩放到0到1之间。然后，使用map函数将这个归一化层应用到训练数据集train_ds和验证数据集val_ds的每个样本上。这样，所有的图像都会被归一化，以便在神经网络中更好地处理。

from tensorflow.keras import layersnormalization_layer=layers.experimental.preprocessing.Rescaling(1./255)train_ds=train_ds.map(lambda x,y:(normalization_layer(x),y))
val_ds=val_ds.map(lambda x,y:(normalization_layer(x),y))

从验证数据集中获取一个批次的图像和标签，然后将第一个图像存储在变量first_image中。接下来，使用numpy库的min和max函数分别计算first_image中的最小值和最大值，并将它们打印出来。这样可以帮助我们了解图像数据的归一化情况，例如是否所有像素值都在0到1之间。

import numpy as npimage_batch,labels_batch=next(iter(val_ds))
first_image=image_batch[0]#查看归一化后的数据
print(np.min(first_image),np.max(first_image))

运行结果如下：

四、构建VGG-16网络

1、VGG优缺点分析

（1）优点：结构简洁，整个网络都使用了同样大小的卷积核尺寸(3x3)和最大池化尺寸（2x2）；

（2）缺点

训练时间过长，调参难度大；
需要的存储容量大，不利于部署。例如存储VGG-16权重值文件的大小为500多MB，不利于安装到嵌入式系统中；

2、自建模型

from tensorflow.keras import layers,models,Input
from tensorflow.keras.models import Model
from tensorflow.keras.layers import Conv2D,MaxPooling2D,Dense,Flatten,Dropoutdef vgg16(nb_classes,input_shape):input_tensor=Input(shape=input_shape)#1st blockx=Conv2D(64,(3,3),activation='relu',padding='same')(input_tensor)x=Conv2D(64,(3,3),activation='relu',padding='same')(x)x=MaxPooling2D((2,2),strides=(2,2))(x)#2nd blockx=Conv2D(128,(3,3),activation='relu',padding='same')(x)x=Conv2D(128,(3,3),activation='relu',padding='same')(x)x=MaxPooling2D((2,2),strides=(2,2))(x)#3rd blockx=Conv2D(256,(3,3),activation='relu',padding='same')(x)x=Conv2D(256,(3,3),activation='relu',padding='same')(x)x=Conv2D(256,(3,3),activation='relu',padding='same')(x)x=MaxPooling2D((2,2),strides=(2,2))(x)#4th blockx=Conv2D(512,(3,3),activation='relu',padding='same')(x)x=Conv2D(512,(3,3),activation='relu',padding='same')(x)x=Conv2D(512,(3,3),activation='relu',padding='same')(x)x=MaxPooling2D((2,2),strides=(2,2))(x)#5th blockx=Conv2D(512,(3,3),activation='relu',padding='same')(x)x=Conv2D(512,(3,3),activation='relu',padding='same')(x)x=Conv2D(512,(3,3),activation='relu',padding='same')(x)x=MaxPooling2D((2,2),strides=(2,2))(x)#full connectionx=Flatten()(x)x=Dense(4096,activation='relu')(x)x=Dense(4096,activation='relu')(x)output_tensor=Dense(nb_classes,activation='softmax',name='predictions')(x)model=Model(input_tensor,output_tensor)return modelmodel=vgg16(len(class_names),(224,224,3))
model.summary()

运行结果如下：

3、网络结构图

结构说明：

13个卷积层（Convolutional Layer），分别用blockX_convX表示；
3个全连接层（Fully connected Layer），分别用fcX与predictions表示；
5个池化层（Pool layer），分别用blockX_pool表示；

VGG-16包含了16个隐藏层（13个卷积层和3个全连接层），故称为VGG-16；

五、编译

在准备对模型进行训练之前，还需要再对其进行一些设置。以下内容是在模型的编译步骤中添加的：

损失函数（loss）：用于衡量模型在训练期间的准确率；
优化器（optimizer）：决定模型如何根据其看到的数据和自身的损失函数进行更新；
指标（metrics）：用于监控训练和测试步骤。以下示例使用了准确率，即被正确分类的图像的比率；

#设置初始学习率
initial_learning_rate=1e-4lr_schedule=tf.keras.optimizers.schedules.ExponentialDecay(initial_learning_rate,decay_steps=30,decay_rate=0.92,staircase=True
)#设置优化器
opt=tf.keras.optimizers.Adam(learning_rate=initial_learning_rate)model.compile(optimizer=opt,loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),metrics=['accuracy'])

六、训练模型

epochs=20history=model.fit(train_ds,validation_data=val_ds,epochs=epochs
)

运行结果如图：

七、可视化结果

acc = history.history['accuracy']
val_acc = history.history['val_accuracy']loss = history.history['loss']
val_loss = history.history['val_loss']epochs_range = range(epochs)plt.figure(figsize=(12, 4))
plt.subplot(1, 2, 1)
plt.plot(epochs_range, acc, label='Training Accuracy')
plt.plot(epochs_range, val_acc, label='Validation Accuracy')
plt.legend(loc='lower right')
plt.title('Training and Validation Accuracy')plt.subplot(1, 2, 2)
plt.plot(epochs_range, loss, label='Training Loss')
plt.plot(epochs_range, val_loss, label='Validation Loss')
plt.legend(loc='upper right')
plt.title('Training and Validation Loss')
plt.show()

运行结果如图：