Image Target Detection Algorithm Compression and Pruning Based on Neural Network

被引:5
|
作者
Sun, Yan [1 ]
Yan, Zheping [1 ]
机构
[1] Harbin Engn Univ, Coll Automat, Harbin 150001, Peoples R China
基金
中国国家自然科学基金;
关键词
Convolutional Neural Network; Target Retrieval; Deep Learning; Algorithm Compression;
D O I
10.2298/CSIS200316007S
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The main purpose of target detection is to identify and locate targets from still images or video sequences. It is one of the key tasks in the field of computer vision. With the continuous breakthrough of deep machine learning technology, especially the convolutional neural network model shows strong Ability to extract image feature in the field of digital image processing. Although the model research of target detection based on convolutional neural network is developing rapidly, but there are still some problems in practical applications. For example, a large number of parameters requires high storage and computational costs in detected model. Therefore, this paper optimizes and compresses some algorithms by using early image detection algorithms and image detection algorithms based on convolutional neural networks. After training and learning, there will appear forward propagation mode in the application of CNN network model, providing the model for image feature extraction, integration processing and feature mapping. The use of back propagation makes the CNN network model have the ability to optimize learning and compressed algorithm. Then research discuss the Faster-RCNN algorithm and the YOLO algorithm. Aiming at the problem of the candidate frame is not significant which extracted in the FasterRCNN algorithm, a target detection model based on the Significant area recommendation network is proposed. The weight of the feature map is calculated by the model, which enhances the saliency of the feature and reduces the background interference. Experiments show that the image detection algorithm based on compressed neural network image has certain feasibility.
引用
收藏
页码:499 / 516
页数:18
相关论文
共 50 条
  • [1] Deep Neural Network Pruning Based Two-Stage Remote Sensing Image Object Detection
    Wang S.-S.
    Wang M.
    Wang G.-Y.
    Dongbei Daxue Xuebao/Journal of Northeastern University, 2019, 40 (02): : 174 - 179
  • [2] Study of Algorithm for Aerial Target Detection Based on Lightweight Neural Network
    Yang, Yumin
    Liao, Yurong
    Ni, Shuyan
    Lin, Cunbao
    2021 IEEE INTERNATIONAL CONFERENCE ON CONSUMER ELECTRONICS AND COMPUTER ENGINEERING (ICCECE), 2021, : 422 - 426
  • [3] ON THE ROLE OF STRUCTURED PRUNING FOR NEURAL NETWORK COMPRESSION
    Bragagnolo, Andrea
    Tartaglione, Enzo
    Fiandrotti, Attilio
    Grangetto, Marco
    2021 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2021, : 3527 - 3531
  • [4] Quantisation and Pruning for Neural Network Compression and Regularisation
    Paupamah, Kimessha
    James, Steven
    Klein, Richard
    2020 INTERNATIONAL SAUPEC/ROBMECH/PRASA CONFERENCE, 2020, : 295 - 300
  • [5] Image Retrieval Algorithm Based on Convolutional Neural Network
    Liu, Hailong
    Li, Baoan
    Lv, Xueqiang
    Huang, Yue
    PROCEEDINGS OF THE 2016 2ND INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE AND INDUSTRIAL ENGINEERING (AIIE 2016), 2016, 133 : 278 - 281
  • [6] Intrusion detection algorithm based on image enhanced convolutional neural network
    Wang, Qian
    Zhao, Wenfang
    Ren, Jiadong
    JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2021, 41 (01) : 2183 - 2194
  • [7] Pruning algorithm of convolutional neural network based on optimal threshold
    Wang, Jianjun
    Liu, Leshan
    Pan, Ximeng
    2020 5TH INTERNATIONAL CONFERENCE ON MATHEMATICS AND ARTIFICIAL INTELLIGENCE (ICMAI 2020), 2020, : 50 - 54
  • [8] Convolutional neural network acceleration algorithm based on filters pruning
    Li H.
    Zhao W.-J.
    Han B.
    Zhejiang Daxue Xuebao (Gongxue Ban)/Journal of Zhejiang University (Engineering Science), 2019, 53 (10): : 1994 - 2002
  • [9] Dim small target detection based on convolutinal neural network in star image
    Xue, Danna
    Sun, Jinqiu
    Hu, Yaoqi
    Zheng, Yushu
    Zhu, Yu
    Zhang, Yanning
    MULTIMEDIA TOOLS AND APPLICATIONS, 2020, 79 (7-8) : 4681 - 4698
  • [10] Dim small target detection based on convolutinal neural network in star image
    Danna Xue
    Jinqiu Sun
    Yaoqi Hu
    Yushu Zheng
    Yu Zhu
    Yanning Zhang
    Multimedia Tools and Applications, 2020, 79 : 4681 - 4698