农作物病虫害识别实操

captain_dong

于 2025-02-07 22:56:05 发布

阅读量409

点赞数 5

文章标签： opencv 计算机视觉机器学习深度学习

本文链接：https://blog.youkuaiyun.com/m0_63064861/article/details/145504265

版权

好的，我将会逐步引导你完成这个农作物病虫害识别项目。我们将从准备环境开始，到最终的模型部署。以下是详细的步骤：

1. 环境准备

首先，你需要安装一些必备的软件和库。我们将使用Python进行开发，并使用TensorFlow进行深度学习模型的训练和部署。

安装Python

从Python官网下载并安装Python。如果你已经安装了Python，可以跳过这一步。

创建虚拟环境

为了确保项目的依赖库不会与其他项目产生冲突，我们使用虚拟环境。

python -m venv crop_disease_env
source crop_disease_env/bin/activate  # 对于Windows，使用 crop_disease_env\Scripts\activate

安装项目依赖

在虚拟环境中，使用pip安装所需的库。

pip install tensorflow opencv-python

2. 项目结构

创建一个新的项目目录，并按照以下结构组织文件：

crop_disease_detection/
├── data/
│   ├── train/
│   ├── test/
├── models/
│   └── crop_disease_model.h5
├── scripts/
│   ├── preprocess_and_train.py
│   └── deploy_and_monitor.py
├── requirements.txt
└── README.md

3. 数据准备

下载PlantVillage数据集并将其解压到data/train/和data/test/目录中。确保数据集结构如下：

data/
├── train/
│   ├── Healthy/
│   ├── Diseased/
├── test/
│   ├── Healthy/
│   ├── Diseased/

4. 编写代码

数据预处理和模型训练

我们将编写一个Python脚本来预处理数据并训练模型。创建文件scripts/preprocess_and_train.py并添加以下内容：

import os
import numpy as np
import tensorflow as tf
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropout

# 数据集路径
dataset_dir = 'data/'

# 图像数据生成器
datagen = ImageDataGenerator(rescale=1.0/255, validation_split=0.2)

# 训练集和验证集生成器
train_generator = datagen.flow_from_directory(
    os.path.join(dataset_dir, 'train'),
    target_size=(128, 128),
    batch_size=32,
    class_mode='categorical',
    subset='training'
)

validation_generator = datagen.flow_from_directory(
    os.path.join(dataset_dir, 'train'),
    target_size=(128, 128),
    batch_size=32,
    class_mode='categorical',
    subset='validation'
)

# 构建模型
model = Sequential([
    Conv2D(32, (3, 3), activation='relu', input_shape=(128, 128, 3)),
    MaxPooling2D((2, 2)),
    Conv2D(64, (3, 3), activation='relu'),
    MaxPooling2D((2, 2)),
    Conv2D(128, (3, 3), activation='relu'),
    MaxPooling2D((2, 2)),
    Flatten(),
    Dense(512, activation='relu'),
    Dropout(0.5),
    Dense(train_generator.num_classes, activation='softmax')
])

# 编译模型
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])

# 训练模型
history = model.fit(
    train_generator,
    epochs=20,
    validation_data=validation_generator
)

# 保存模型
model.save('models/crop_disease_model.h5')

模型部署和实时监测

我们将编写另一个Python脚本来部署模型并进行实时监测。创建文件scripts/deploy_and_monitor.py并添加以下内容：

import cv2
import numpy as np
import tensorflow as tf
from tensorflow.keras.models import load_model

# 加载训练好的模型
model = load_model('models/crop_disease_model.h5')

# 类别标签
class_labels = ['Healthy', 'Diseased']

# 初始化摄像头
cap = cv2.VideoCapture(0)

while True:
    # 读取摄像头帧
    ret, frame = cap.read()
    if not ret:
        break
    
    # 预处理图像
    img = cv2.resize(frame, (128, 128))
    img = np.expand_dims(img, axis=0)
    img = img / 255.0
    
    # 进行预测
    predictions = model.predict(img)
    predicted_class = class_labels[np.argmax(predictions)]
    
    # 在图像上显示预测结果
    cv2.putText(frame, predicted_class, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
    cv2.imshow('Crop Disease Monitor', frame)
    
    # 按下 'q' 键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放资源
cap.release()
cv2.destroyAllWindows()