Image target detection method and system based on lightweight neural network model
A neural network model and object detection technology, which is applied in the field of image object detection, can solve problems such as high cost of high-performance GPU, inapplicable model deployment, and inability to move, so as to meet real-time performance and accuracy, reduce model size, and improve accuracy Effect
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment approach
[0037] In the case of no conflict, the embodiments and the features in the embodiments of the present invention can be combined with each other.
[0038] The current mainstream research direction of the target detection algorithm based on deep learning is to continuously reduce the weight of the neural network model, and continuously improve the operation ability of the model on small devices and mobile terminals, so that the model can be better applied to actual production and life. more socioeconomic benefits.
Embodiment 1
[0040] This embodiment discloses an image target detection method based on a lightweight neural network model. This implementation example takes smoking behavior detection as an example for illustration, and of course it can also be applied to image detection of other target behaviors.
[0041] Include the following steps:
[0042] S1: Make a data set for deep learning training and testing, and divide and process the entire data set;
[0043] S2: Configure the python and pytorch programming environment for neural network model training and testing;
[0044] S3: Construct the backbone network and feature fusion network required to implement the yolov5 target detection algorithm, where the backbone network is used to extract useful features in the image to be detected, and the feature fusion network is used to strengthen the useful features extracted by the backbone network and output the image to be detected The final feature map of ;
[0045] S4: Define the loss function of ...
Embodiment 2
[0120] The purpose of this embodiment is to provide an image target detection system based on a lightweight neural network model, including:
[0121] The data input module is configured to: input the path of the picture or video to be detected;
[0122] The target detection module is configured to: use the lightweight neural network model to calculate the relative confidence of all classifications in the received current image, and select the highest confidence to obtain the final recognition frame and draw it in the original picture to complete the detection process.
[0123] The present invention relates to neural network, deep learning, machine vision, and target detection technology. It mainly uses the latest single-stage target detection algorithm yolov5, adjusts the width (width_multiple) and depth (depth_multiple) of the network structure, and adjusts the backbone network (backbone ) to improve Conv and CSPNet, effectively reducing the complexity of the model while ensu...
PUM
Abstract
Description
Claims
Application Information
- R&D Engineer
- R&D Manager
- IP Professional
- Industry Leading Data Capabilities
- Powerful AI technology
- Patent DNA Extraction
Browse by: Latest US Patents, China's latest patents, Technical Efficacy Thesaurus, Application Domain, Technology Topic, Popular Technical Reports.
© 2024 PatSnap. All rights reserved.Legal|Privacy policy|Modern Slavery Act Transparency Statement|Sitemap|About US| Contact US: help@patsnap.com