一种基于小波注意力融合机制的蔗尾计数方法

本发明涉及图像处理领域，尤其是种一基于小波注意力融合机制的蔗尾计数方法，包括下述步骤：通过摄像头获取蔗尾的原始特征图；通过小波卷织对所述原始特征图进行处理，以获得子带特征图；通过所述注意力权重对所述子带特征图进行加权特征增强，以获得加权子带特征图；通过空间注意力权重及通道注意力权重对所述原始特征图进行处理，以获得加权特征图；通过解码器对所述最终特征图进行上采样，且生成密度图，根据密度图获得蔗尾的数量；收集甘蔗尾部的图像数据并构建数据集；构建初始计数模型，且通过所述数据集进行模型训练，以获得优化计数模型。本发明能够同步提升颜色相似目标的特征区分度、高频细节保留能力及遮挡场景下的计数鲁棒性。