基于区域亮度自适应校正的茶叶嫩芽检测模型
2021-02-19方梦瑞武传宇贺盈磊钟小玉
吕 军,方梦瑞,姚 青,武传宇,贺盈磊,边 磊,钟小玉
基于区域亮度自适应校正的茶叶嫩芽检测模型
吕 军1,方梦瑞1,姚 青1※,武传宇2,贺盈磊2,边 磊3,钟小玉4
(1. 浙江理工大学信息学院,杭州 310018;2. 浙江理工大学机械与自动控制学院,杭州 310018;3. 中国农业科学院茶叶研究所,杭州 310008;4. 浙江茗皇天然食品开发股份有限公司,杭州 310030)
自然光照下不同时间采集的茶叶图像存在亮度不均的现象。由于高亮度图像对比度差且嫩芽特征显著性弱,造成高亮度图像中存在较多嫩芽的漏检。针对现有茶叶嫩芽图像自动检测方法对光照变化的敏感性,该研究提出一种基于区域亮度自适应校正的茶叶嫩芽检测模型。首先,对不同时间采集的龙井43茶叶图像进行灰度化;然后,计算灰度图的平均灰度(Average Gray,AG)值,对AG值在[170,230]的高亮度图像进行不同尺寸的分块处理和局部区域伽马亮度自适应校正;最后,在相同的训练集和测试集训练多个深度学习检测模型。测试结果表明,基于YOLOv5+CSPDarknet53的检测模型比SSD+VGG16、Faster RCNN+VGG16、YOLOv3+Darknet53和YOLOv4+CSPDarknet53模型具有更优的嫩芽检测性能,精确率和召回率分别为88.2%和82.1%。对YOLOv5检测结果进行检测抑制,有效避免了同一目标被多次框选的冗余现象。[30,90)和[90,170)亮度区间内嫩芽图像具有较强的显著性特征和较高的检测精度与召回率。相较于AG值在[170,230]的高亮度原始图像的检测结果,对高亮度图像进行2×3分块和局部区域亮度自适应校正后,YOLOv5嫩芽检测召回率提高了19.2个百分点。对不同光照条件下采集的茶叶图像进行测试,基于区域亮度自适应校正的茶叶嫩芽YOLOv5检测模型获得了92.4%的检测精度和90.4%的召回率。……
