人工智能在病虫害图像识别中的效果：从理论到代码实战

1. 行业痛点与研究动机

• 传统植保痛点：人工巡田耗时、误诊率高；化学农药滥用导致抗药性与生态破坏。
• AI 介入的价值：基于手机或无人机拍摄的叶片/果实图像，实时给出病虫害种类与置信度，指导精准施药。
• 技术挑战：
  1. 数据类别极不均衡（健康样本远多于病害）；
  2. 细粒度差异小（同类病害不同亚型）；
  3. 田间光照、背景复杂。

本文将围绕“如何构建一个可解释、轻量且易落地的病虫害识别系统”展开：

• 数据集：PlantVillage + 自建田间数据；
• 模型：EfficientNet-B3 + 迁移学习 + 类别平衡损失；
• 可解释性：Grad-CAM 热力图；
• 部署：TensorFlow Lite 量化到手机端（Android）。

2. 数据集构建与标注策略

2.1 数据源

2.2 标注流程

1. 使用 CVAT进行多边形分割 → 生成 mask.png；
2. 按 8:1:1 划分训练/验证/测试，分层抽样保证每类比例一致；
3. 用 albumentations 做离线数据增强（随机亮度、HSV、CutMix）。

3. 模型设计：EfficientNet-B3 + 类别平衡损失

3.1 网络选择

• EfficientNet-B3 在 ImageNet 上 Top-1 81.6%，参数量仅 12M；
• 输入分辨率 300×300，适配手机端实时推断。

3.2 类别不平衡的两种策略

实验表明，Focal Loss + 0.25 γ 在宏平均 F1 上提升 3.8%。

4. 代码实战：从训练到推断

4.1 环境准备

conda create -n plantai python=3.10
pip install tensorflow==2.15 albumentations==1.3 grad-cam==1.4.5

4.2 数据管道（TensorFlow 2.x）

import tensorflow as tf, albumentations as A, cv2, numpy as np

def aug_fn(image):
    transform = A.Compose([
        A.RandomRotate90(),
        A.ColorJitter(0.2,0.2,0.2,0.2,p=0.8),
        A.CutMix(p=0.5),
    ])
    return transform(image=image)['image']

def parse_path(path, label):
    img = tf.io.read_file(path)
    img = tf.image.decode_jpeg(img, channels=3)
    img = tf.numpy_function(aug_fn, [img], tf.uint8)
    img = tf.image.resize(img, [300,300])
    return img/255., label

train_ds = (tf.data.Dataset
            .from_tensor_slices((paths, labels))
            .shuffle(2048)
            .map(parse_path, num_parallel_calls=tf.data.AUTOTUNE)
            .batch(32)
            .prefetch(tf.data.AUTOTUNE))

4.3 模型构建与训练

from tensorflow.keras.applications import EfficientNetB3
from tensorflow.keras import layers, models

base = EfficientNetB3(include_top=False, weights='imagenet', input_shape=(300,300,3))
base.trainable = False  # 先冻结 backbone

x = layers.GlobalAveragePooling2D()(base.output)
x = layers.Dropout(0.3)(x)
outputs = layers.Dense(num_classes, activation='softmax')(x)
model = models.Model(base.input, outputs)

model.compile(
    optimizer=tf.keras.optimizers.Adam(1e-3),
    loss=tf.keras.losses.CategoricalFocalCrossentropy(alpha=0.25, gamma=2.0),
    metrics=['accuracy', tf.keras.metrics.F1Score(average='macro')]
)

# 解冻最后 60 层做微调
for layer in base.layers[-60:]:
    layer.trainable = True

model.fit(train_ds,
          epochs=20,
          validation_data=val_ds,
          callbacks=[
              tf.keras.callbacks.EarlyStopping(patience=3, restore_best_weights=True),
              tf.keras.callbacks.ReduceLROnPlateau(factor=0.5, patience=2)
          ])

4.4 评估结果

5. 可解释性：Grad-CAM 热力图

from tensorflow.keras.models import Model
from gradcam import GradCAM

# 取倒数第二个卷积层
conv_layer = model.get_layer('top_conv')  # EfficientNet 顶层
gradcam = GradCAM(model, conv_layer)

heatmap = gradcam.compute_heatmap(img_array, classIdx=np.argmax(pred))
heatmap = cv2.resize(heatmap, (300,300))
overlay = heatmap * 0.4 + img_array[0]
cv2.imwrite('explain.jpg', overlay)

下图展示了模型在“番茄晚疫病”样本上的注意力区域，可见模型主要聚焦在病变边缘而非背景泥土，验证了可解释性。

6. 端侧部署：TensorFlow Lite 量化

6.1 量化训练

converter = tf.lite.TFLiteConverter.from_keras_model(model)
converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.representative_dataset = representative_data_gen  # 100 张样本
tflite_model = converter.convert()
open('plant_model.tflite','wb').write(tflite_model)

6.2 Android 端集成

val tflite = Interpreter(loadModelFile("plant_model.tflite"))
val input = preprocess(bitmap)  // 归一化到 [0,1]
val output = Array(1){FloatArray(numClasses)}
tflite.run(input, output)
val prob = output[0]
val label = labels[prob.indexOfFirst { it == prob.max() }]

APK 包体仅 5.4 MB，推断延迟 38 ms，满足实时拍照识别需求。

7. 结果讨论与落地建议

7.1 误差分析

• 混淆矩阵显示，“早疫病”与“晚疫病”误分率 4.2%，原因：二者早期病斑颜色相似；
• 阴影遮挡导致的漏检可通过 多光谱成像 进一步缓解。

7.2 业务闭环

1. 农户拍照 → 手机 App 实时识别 → 推荐农药与剂量；
2. 后台收集误报图像 → 人工二次标注 → 增量训练（每季度更新模型）。

8. 结语与展望

本文完整呈现了从数据到部署的全过程，验证了 AI 在病虫害识别中的 高精度（96.7% Top-1）、低延迟（<40 ms）、可解释性。未来方向：

• 引入 SAM（Segment Anything Model） 做实例级病斑分割，实现“病斑面积”量化；
• 融合 气象数据 + 时序模型，预测病害爆发概率，实现 预防式植保。