A Lightweight CNN Model Based on GhostNet

Abstract

The existing deep learning models have problems such as large weight parameters and slow inference speed of equipment. In practical applications such as fire detection, they often cannot be deployed on equipment with limited resources due to the huge amount of parameters and low efficiency. In response to this problem, this paper proposes a lightweight smoke detection model based on the convolutional attention mechanism module. The model is based on the YOLOv5 lightweight framework. The backbone network draws on the GhostNet design idea, replaces the CSP structure of the FPN and head layers with the GhostBottleNeck module, adds a convolutional attention mechanism module to the backbone network layer, and uses the CIoU loss function to improve the regression accuracy. Using YOLOv5s as the benchmark model, the parameter amount of the proposed lightweight neural network model is 2.75 M, and the floating-point calculation amount is 2.56 G, which is much lower than the parameter amount and calculation amount of the benchmark model. Tested on the public fire dataset, compared with the traditional deep learning algorithm, the model proposed in the paper has better detection performance and the detection speed is significantly better than the benchmark model. Tested under the unquantized simulator, the speed of the proposed model to detect a single picture is 60 ms, which can meet the requirements of real-time engineering applications.

A Fine-grained Attributes Recognition Model for Clothing Based on Improved the CSPDarknet and PAFPN Network

“…This detailed categorization provides the basis for further research and provides us with a strong support for deeper understanding of apparel consumer behavior and market trends. The GhostConv module is a convolution module in the GhostNet [25] network that replaces normal convolution. The structure of the GhostConv module is shown in Fig.…”

Performance Assessment of DL Model in Iraq for Covid-19 Patient Severity Prediction from X-Ray Scan Images

“…Ghost module that has been presented shrinks the size of the model and lowers computational costs without significantly sacrificing accuracy [19].…”