YOLOv9农业无人机应用：作物密度统计部署实战

1. 引言

1.1 农业智能化的迫切需求

现代农业正加速向数字化、智能化转型。在精准农业场景中，作物密度统计是田间管理的关键环节，直接影响播种规划、施肥决策与产量预估。传统人工调查方式效率低、成本高，难以满足大范围农田的实时监测需求。随着无人机航拍技术的普及，结合深度学习目标检测算法实现自动化作物计数，已成为行业主流解决方案。

1.2 YOLOv9的技术优势

YOLOv9 是 YOLO 系列的最新演进版本，由 WongKinYiu 团队于 2024 年提出，其核心创新在于引入可编程梯度信息（Programmable Gradient Information, PGI）和广义高效层聚合网络（Generalized Efficient Layer Aggregation Networks, GELAN），显著提升了小目标检测精度与模型收敛速度。相比 YOLOv5/v8，YOLOv9 在保持轻量化的同时，在 MS-COCO 数据集上实现了更高的 mAP 和更低的参数量，特别适合部署在边缘设备或无人机端进行实时推理。

1.3 实战目标与价值

本文聚焦于将 YOLOv9 应用于农业无人机图像中的作物密度统计任务，基于官方训练与推理镜像完成从数据准备到模型部署的全流程实践。通过本教程，读者将掌握： - 如何使用预置镜像快速搭建 YOLOv9 开发环境 - 针对农作物图像的数据标注与格式转换方法 - 模型微调训练的关键参数配置 - 在真实航拍图像上执行高效推理并生成密度热力图

2. 环境准备与镜像使用

2.1 镜像环境说明

本文所用镜像基于 YOLOv9 官方代码库构建，预装了完整的深度学习开发环境，集成训练、推理及评估所需的所有依赖，支持开箱即用。

组件	版本
核心框架	PyTorch 1.10.0
CUDA 版本	12.1
Python 版本	3.8.5
Torchvision	0.11.0
Torchaudio	0.10.0
CUDAToolkit	11.3
OpenCV	opencv-python
其他依赖	numpy, pandas, matplotlib, tqdm, seaborn

代码位于/root/yolov9目录下，权重文件yolov9-s.pt已预下载至该路径。

2.2 激活环境

启动容器后，默认处于base环境，需手动激活yolov9虚拟环境：

conda activate yolov9

2.3 进入项目目录

切换至 YOLOv9 主目录以执行后续命令：

cd /root/yolov9

3. 数据准备与格式规范

3.1 农业图像采集建议

为确保检测效果，建议使用无人机在固定高度（如 10–20 米）垂直拍摄 RGB 图像，分辨率不低于 1920×1080。拍摄时间应选择光照均匀的上午或下午，避免强烈阴影干扰。

3.2 标注工具与流程

推荐使用 LabelImg 或 CVAT 对作物个体进行框选标注。每株作物作为一个独立对象，类别统一设为“crop”。

提示：对于密集种植场景，可采用点标注（point annotation）后转换为小尺寸边界框（如 16×16 像素），提升标注效率。

3.3 YOLO 格式要求

YOLO 要求标注文件为.txt格式，每行表示一个对象，格式如下：

<class_id> <x_center> <y_center> <width> <height>

所有坐标归一化到 [0, 1] 区间。图像与标签文件同名，分别存放于images/和labels/文件夹。

3.4 数据集划分与配置

创建data.yaml文件定义数据结构：

train: /path/to/train/images val: /path/to/val/images nc: 1 names: ['crop']

请根据实际路径修改train和val字段。

4. 模型训练：从预训练到微调

4.1 训练命令详解

使用以下命令启动单卡训练：

python train_dual.py \ --workers 8 \ --device 0 \ --batch 64 \ --data data.yaml \ --img 640 \ --cfg models/detect/yolov9-s.yaml \ --weights '' \ --name yolov9-s-crop \ --hyp hyp.scratch-high.yaml \ --min-items 0 \ --epochs 50 \ --close-mosaic 15

参数解析：

--workers 8：数据加载线程数，视 CPU 性能调整
--batch 64：批量大小，显存不足时可降至 32 或 16
--img 640：输入图像尺寸，作物检测建议不小于 640
--weights ''：使用空字符串表示从头训练；若继续预训练，则填'./yolov9-s.pt'
--hyp hyp.scratch-high.yaml：适用于从零开始训练的高增益超参配置
--close-mosaic 15：在最后 15 轮关闭 Mosaic 数据增强，提升收敛稳定性

4.2 训练过程监控

训练日志和检查点保存在runs/train/yolov9-s-crop目录下，包含： -weights/best.pt：验证集 mAP 最高的模型 -weights/last.pt：最后一轮保存的模型 -results.csv：各轮次指标记录 -confusion_matrix.png：分类混淆矩阵

建议定期查看val_batch0_labels.jpg和val_batch0_pred.jpg，直观评估检测质量。

5. 模型推理与作物密度分析

5.1 单图推理测试

使用训练好的模型对新图像进行预测：

python detect_dual.py \ --source './data/images/test_field.jpg' \ --img 640 \ --device 0 \ --weights 'runs/train/yolov9-s-crop/weights/best.pt' \ --name crop_density_detect \ --conf-thres 0.5 \ --iou-thres 0.45

输出结果保存在runs/detect/crop_density_detect目录中，包含原图叠加检测框的可视化图像。

5.2 批量推理与计数统计

若需处理整个农田的多张航拍图，可指定目录作为输入源：

python detect_dual.py \ --source '/path/to/aerial_images/' \ --img 640 \ --device 0 \ --weights 'runs/train/yolov9-s-crop/weights/best.pt' \ --name batch_crop_count \ --save-txt \ --save-conf

启用--save-txt后，每张图像对应一个.txt文件，记录所有检测结果（格式同训练标签）。可通过脚本自动统计每图作物数量：

import os def count_crops_in_dir(label_dir): counts = {} for file in os.listdir(label_dir): if file.endswith('.txt'): with open(os.path.join(label_dir, file), 'r') as f: lines = f.readlines() counts[file] = len(lines) return counts counts = count_crops_in_dir('runs/detect/batch_crop_count/labels') print(counts)

5.3 密度热力图生成

结合 GPS 信息（若有），可将作物数量映射到地理坐标，生成农田密度热力图。示例代码如下：

import matplotlib.pyplot as plt import seaborn as sns import numpy as np # 模拟网格化计数数据 (例如 10x10 网格) grid_counts = np.random.poisson(lam=15, size=(10, 10)) # 替换为真实数据 plt.figure(figsize=(8, 8)) sns.heatmap(grid_counts, annot=True, fmt="d", cmap="YlGnBu", cbar_kws={'label': 'Crop Count'}) plt.title("Crop Density Heatmap") plt.xlabel("Grid X") plt.ylabel("Grid Y") plt.tight_layout() plt.savefig("crop_density_heatmap.png", dpi=150) plt.show()