基于全局-局部注意机制的天气分类算法

摘要

天气分类算法是户外交通视频分析以及计算机视觉应用的重要组成部分,针对天气分类在多样性和准确性等方面的要求,提出了一种基于全局-局部注意机制的多类天气分类算法,对户外交通图像进行天气分类。利用基础网络ResNet获得图像的特征图,构建一种改进的通道注意模型来引入全局注意分支,通过堆叠一系列的卷积层、非线性层和池化层得到能够充分表征全局信息的特征,进一步结合局部注意分支网络,通过学习得到与图像语义信息最具鉴别性的局部特征,将增强之后的特征图用于天气分类模型。