Image Target Detection Algorithm Compression and Pruning Based on Neural Network

Yan Sun1 and Zheping Yan1

  1. College of Automation, Harbin Engineering University
    Harbin 150001, China
    {aidenby, feiyunsy1213}@163.com

Abstract

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 Faster-RCNN 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.

Key words

Convolutional Neural Network,Target Retrieval, Deep Learning, Algorithm Compression

Digital Object Identifier (DOI)

https://doi.org/10.2298/CSIS200316007S

Publication information

Volume 18, Issue 2 (April 2021)
Special Issue on Emerging Services in the Next-Generation Web: Human Meets Artificial Intelligence
Year of Publication: 2021
ISSN: 1820-0214 (Print) 2406-1018 (Online)
Publisher: ComSIS Consortium

Full text

DownloadAvailable in PDF
Portable Document Format

How to cite

Sun, Y., Yan, Z.: Image Target Detection Algorithm Compression and Pruning Based on Neural Network. Computer Science and Information Systems, Vol. 18, No. 2, 499–516. (2021), https://doi.org/10.2298/CSIS200316007S