1. 数据采集:
· 地点:河南新乡小麦田
· 设备:iPhone XR
· 数据量:1415张图像(条锈病600张,白粉病815张)
2. 数据预处理:
· 图像归一化至640×640像素
· 使用LabelImg标注,转换为YOLO格式
· 数据增强:亮度调整、高斯模糊、添加噪声等
(条锈病害图像数据增强效果示例如图5所示)
3. 模型改进(CS-YOLO)(见图6-13):
· 替换C2f中的Bottleneck为FasterNet Block(减少计算量)
· 引入GSConv和VoV-GSCSP模块(提升特征融合能力)
4. 训练设置:
(环境见表1)
· 框架:PyTorch 1.13,Python 3.8
· 优化器:SGD,学习率0.01,动量0.937
· 迭代次数:200
5. 评价指标:
· mAP@0.5
· 浮点运算量(FLOPs)
· 模型内存占用量