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 条
  • [21] Optimization of the Mixed Gas Detection Method Based on Neural Network Algorithm
    Li, Xiulei
    Cao, Juexian
    Xu, Wangping
    Guo, Jiayi
    ACS SENSORS, 2023, 8 (02) : 822 - 828
  • [22] Cancer Detection Based on Image Classification by Using Convolution Neural Network
    Shah, Mohammad Anas
    Nour, Abdala
    Ngom, Alioune
    Rueda, Luis
    BIOINFORMATICS AND BIOMEDICAL ENGINEERING (IWBBIO 2020), 2020, 12108 : 275 - 286
  • [23] Image Forgery Detection Based on the Convolutional Neural Network
    Feng Guorui
    Wu Jian
    ICMLC 2020: 2020 12TH INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND COMPUTING, 2018, : 266 - 270
  • [24] Convolutional Neural Network Compression via Dynamic Parameter Rank Pruning
    Sharma, Manish
    Heard, Jamison
    Saber, Eli
    Markopoulos, Panagiotis
    IEEE ACCESS, 2025, 13 : 18441 - 18456
  • [25] Target Detection of Hyperspectral Image Based on Convolutional Neural Networks
    Liu, Xuefeng
    Wang, Congcong
    Sun, Qiaoqiao
    Fu, Min
    2018 37TH CHINESE CONTROL CONFERENCE (CCC), 2018, : 9255 - 9260
  • [26] Model Compression Based on Differentiable Network Channel Pruning
    Zheng, Yu-Jie
    Chen, Si-Bao
    Ding, Chris H. Q.
    Luo, Bin
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (12) : 10203 - 10212
  • [27] Improved image super-resolution algorithm based on convolutional neural network
    Xiao J.
    Liu E.
    Zhu L.
    Lei J.
    1600, Chinese Optical Society (37):
  • [28] Pruning- and Quantization-Based Compression Algorithm for Number of Mixed Signals Identification Network
    Shen, Weiguo
    Wang, Wei
    Zhu, Jiawei
    Zhou, Huaji
    Wang, Shunling
    ELECTRONICS, 2023, 12 (07)
  • [29] Deep Neural Network Compression by In-Parallel Pruning-Quantization
    Tung, Frederick
    Mori, Greg
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2020, 42 (03) : 568 - 579
  • [30] Group Pruning with Group Sparse Regularization for Deep Neural Network Compression
    Wu, Chenglu
    Pang, Wei
    Liu, Hao
    Lu, Shengli
    2019 IEEE 4TH INTERNATIONAL CONFERENCE ON SIGNAL AND IMAGE PROCESSING (ICSIP 2019), 2019, : 325 - 329