基于卷积神经网络的灾难场景图像分类
DOI:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

国家自然科学基金(61601176)


Image Classification of Disaster Scenes Based on Convolutional Neural Network
Author:
Affiliation:

Fund Project:

The National Natural Science Foundation of China (General Program, Key Program, Major Research Plan)

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了对灾难场景图像进行快速分析和识别,提出了一种基于多分辨率卷积神经网络和残差注意力机制(Attention Module)相结合的图像分类模型。首先,对灾难场景数据集进行预处理,在相同类型的条件下将其随机划分为训练集和测试集。基于改进的卷积神经网络(convolutional neural network,CNN),提取训练集的图像特征。使用训练集图片的特征进行训练,并且在测试集图片上实现分类测试。选取DenseNet、Xception和MobileNetV2三种模型,以灾难场景数据集(Disaster_Data_Scenes)为数据集进行实验验证。结果表明:改进的Xception和MobileNetV2网络在灾难场景数据集上进行的图像分类实验测试,比原网络精度分别提升了4.56%,3.04%。其中改进的DenseNet网络比原网络模型精度分别提升9.13%,17.88%,10.27%。可见改进的卷积神经网络模型的分类精度得到有效提高。

    Abstract:

    In order to quickly analyze and recognize disaster scene images, an image classification model based on the combination of multi-resolution convolutional neural network and the Attention Module was proposed.First, the disaster scenario data set was preprocessed and randomly divided into training set and test set under the same type of conditions.The image characteristics of the training set were extracted by an improved convolutional neural network (CNN).The characteristics of the training set images were used for training, and the classification test was implemented on the test set images.The three models of DenseNet, Xception and MobileNetV2 were selected and the disaster scene data set (Disaster_Data_Scenes) was used for experimental verification.The results show that the improved Xception and MobileNetV2 network improved the accuracy of image classification by 4.56% and 3.04%, respectively, compared with the original network.Among them, the accuracy of the improved DenseNet network is 9.13%, 17.88% and 10.27% higher than that of the original network model.It is concluded that the classification accuracy of the improved convolutional neural network model is effectively improved.

    参考文献
    相似文献
    引证文献
引用本文

王改华,郭钊,周志刚,等. 基于卷积神经网络的灾难场景图像分类[J]. 科学技术与工程, 2021, 21(17): 7217-7223.
Wang Gaihua, Guo Zhao, Zhou Zhigang, et al. Image Classification of Disaster Scenes Based on Convolutional Neural Network[J]. Science Technology and Engineering,2021,21(17):7217-7223.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2020-10-23
  • 最后修改日期:2021-01-28
  • 录用日期:2021-02-23
  • 在线发布日期: 2021-07-02
  • 出版日期:
×
律回春渐,新元肇启|《科学技术与工程》编辑部恭祝新岁!
亟待确认版面费归属稿件,敬请作者关注