花朵分类(一)

2021-07-29 15:05:36 1.8k 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

深度学习之基于Matlab GoogleNet网络的花朵分类识别一、项目背景花朵分类识别是计算机视觉领域的一个重要应用，具有广泛的应用前景，如园艺、植物学、生态保护等。然而，传统的花朵分类方法往往依赖于人工识别，存在效率低、准确性不高等问题。随着深度学习技术的不断发展，越来越多的研究者开始尝试使用深度学习算法来解决花朵分类问题。本项目基于Matlab平台和GoogleNet网络，旨在开发一个高效、准确的花朵分类识别系统。二、项目目标本项目的主要目标是利用深度学习技术，特别是GoogleNet网络，结合Matlab平台，实现花朵图像的自动分类识别。阅读详情

本次教程的目的是带领大家学会基本的花朵图像分类

首先我们来介绍下数据集，该数据集有5种花，一共有3670张图片，分别是daisy、dandelion、roses、sunflowers、tulips，数据存放结构如下所示

我们可以展示下roses的几张图片

接下来我们需要加载数据集，然后对数据集进行划分，最后形成训练集、验证集、测试集，注意此处的验证集是从训练集切分出来的，比例是8:2

对数据进行探索的时候，我们发现原始的像素值是0-255，为了模型训练更稳定以及更容易收敛，我们需要标准化数据集，一般来说就是把像素值缩放到0-1，可以用下面的layer来实现

normalization_layer = tf.keras.layers.experimental.preprocessing.Rescaling(1./255)

为了使训练的时候I/O不成为瓶颈，我们可以进行如下设置

AUTOTUNE = tf.data.AUTOTUNE

train_ds = train_ds.cache().prefetch(buffer_size=AUTOTUNE)
val_ds = val_ds.cache().prefetch(buffer_size=AUTOTUNE)

下一步就是模型搭建，然后对模型进行训练

num_classes = 5

model = tf.keras.Sequential([
  tf.keras.layers.experimental.preprocessing.Rescaling(1./255),
  tf.keras.layers.Conv2D(32, 3, activation='relu'),
  tf.keras.layers.MaxPooling2D(),
  tf.keras.layers.Conv2D(32, 3, activation='relu'),
  tf.keras.layers.MaxPooling2D(),
  tf.keras.layers.Conv2D(32, 3, activation='relu'),
  tf.keras.layers.MaxPooling2D(),
  tf.keras.layers.Flatten(),
  tf.keras.layers.Dense(128, activation='relu'),
  tf.keras.layers.Dense(num_classes)
])

model.compile(
  optimizer='adam',
  loss=tf.losses.SparseCategoricalCrossentropy(from_logits=True),
  metrics=['accuracy'])

model.fit(
  train_ds,
  validation_data=val_ds,
  epochs=3
)