使用Python+OpenCV+Keras实现植物幼苗分类(附python演练)

介绍

本文将学习和构建一个 CNN 模型，以从图像中对幼苗的种类进行分类。该数据集有12组图像，我们的最终目的是从图像中对植物物种进行分类。

如果你想了解有关数据集的更多信息，请查看此链接：https://www.kaggle.com/c/plant-seedlings-classification/data。

我们将执行多个步骤，例如导入库和模块、读取图像并调整它们的大小、图像清理、图像预处理、模型构建、模型训练、减少过度拟合，最后对测试数据集进行预测。

使用深度学习解决数独，查看这里：https://www.analyticsvidhya.com/blog/2021/05/solving-sudoku-from-image-using-deep-learning-with-python-code/

目录

• 问题陈述

• 导入库

• 获取数据并调整图像大小

• 清理图像并去除背景

• 将标签转换为数字

• 定义我们的模型并拆分数据集

• 防止过拟合

• 定义卷积神经网络

• 将 CNN 拟合到数据上

• 混淆矩阵

• 获得预测

问题陈述

该数据集由奥尔胡斯大学信号处理小组提供。这是一个典型的图像识别问题陈述。我们提供了一个图像数据集，其中包含处于不同生长阶段的植物照片。每张照片都有其唯一的 ID 和文件名。

该数据集包含来自 12 个植物物种的 960 种独特植物。最终目标是构建一个能够从照片中确定植物种类的分类器。

物种列表

• Black-grass

• Charlock

• Cleavers

• Common Chickweed

• Common wheat

• Fat Hen

• Loose Silky-bent

• Maize

• Scentless Mayweed

• Shepherds Purse

• Small-flowered Cranesbill

• Sugar beet

导入库

首先导入所有必要的库以供我们进一步分析。我们将使用 NumPy、Pandas、matplotlib、OpenCV、Keras 和 sci-kit-learn。

检查以下命令以导入所有必需的库

import numpy as np # MATRIX OPERATIONS
import pandas as pd # EFFICIENT DATA STRUCTURES
import matplotlib.pyplot as plt # GRAPHING AND VISUALIZATIONS
import math # MATHEMATICAL OPERATIONS
import cv2 # IMAGE PROCESSING - OPENCV
from glob import glob # FILE OPERATIONS
import itertools
# KERAS AND SKLEARN MODULES
from keras.utils import np_utils
from keras.preprocessing.image import ImageDataGenerator
from keras.models import Sequential
from keras.layers import Dense
from keras.layers import Dropout
from keras.layers import Flatten
from keras.layers.convolutional import Conv2D
from keras.layers.convolutional import MaxPooling2D
from keras.layers import BatchNormalization
from keras.callbacks import ModelCheckpoint,ReduceLROnPlateau,CSVLogger
from sklearn import preprocessing
from sklearn.model_selection import train_test_split
from sklearn.metrics import confusion_matrix
# GLOBAL VARIABLES
scale = 70
seed = 7

获取数据并调整图像大小

为了训练我们的模型，我们需要先读取数据。我们的数据集有不同大小的图像，因此我们将调整图像的大小。读取数据并调整其大小只需一步即可完成。查看以下代码以获取有关如何执行不同操作的完整信息。

path_to_images =