图像分类方法、系统、计算机设备及存储介质

作者：王伟凝; 李乐敏; 郭沛榕; 谭燕; 石红霞

来源：2020-09-14, 中国, ZL202010959142.6.

本发明公开了一种图像分类方法、系统、计算机设备及存储介质,所述方法包括：构造高效全局上下文模块；搭建图像分类网络,所述图像分类网络包括残差网络以及插入残差网络中的四个高效全局上下文模块；获取训练集,所述训练集包括多张训练图像；利用训练集对图像分类网络进行训练,使用交叉熵损失函数,采用随机梯度下降法对图像分类网络进行迭代优化,得到图像分类模型；获取待分类图像；将待分类图像输入图像分类模型中,得到分类预测结果。本发明通过在残差网络中插入高效全局上下文模块,以搭建图像分类网络,有效地捕获全局信息以及通道依赖信息,提高了图像分类网络的分类识别能力,达到更好的图像分类效果。

• 单位
  华南理工大学