Household garbage real-time detection method and device, electronic equipment and medium
A technology for real-time detection of domestic garbage, applied in the field of computer vision, can solve problems such as high proportion of organic waste, poor work quality, high water content, etc., and achieve the effect of improving detection accuracy, improving accuracy, and excellent detection accuracy
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0047] Such as figure 1 As shown, a real-time detection method for domestic garbage based on attention mechanism combination proposed in this embodiment includes the following steps:
[0048] S1: Obtain a garbage image detection data set, and divide the garbage image detection data set into a training set, a verification set and a test set;
[0049] In this embodiment, the garbage image detection data set is obtained by collecting garbage images online, capturing household garbage in videos, and taking photos, etc., and marking all household garbage in the images according to their product names to obtain a total of 13 types of garbage, and 13 is divided into four categories, namely: domestic garbage, recyclable garbage, hazardous garbage and other garbage, establish a garbage detection data set containing multiple categories, and divide the data set into training set, verification set and test set; Specific steps are as follows:
[0050] Step S1-1: define the type and size ...
Embodiment 2
[0076] The present invention also provides a cultural relic image color restoration device, which includes:
[0077] A preprocessing module, configured to obtain a garbage image detection data set, and divide the garbage image detection data set into a training set, a verification set and a test set;
[0078] The training module is used to input the rubbish images in the training set and the verification set into the improved YOLOv5s network model respectively and carry out iterative training through GPU, and obtain the optimal weight of the improved YOLOv5s network model after training; the improved YOLOv5s network model The YOLOv5s network model is improved on the YOLOv5s network model, specifically: a lightweight feedforward convolutional attention module CBAM is introduced after the front-end Focus module of the backbone network of the YOLOv5s network model, and SE- Net channel attention module;
[0079] The garbage detection module is used to load the optimal weight into...
Embodiment 3
[0091] The present invention also provides an electronic device, which is characterized in that the electronic device includes: at least one processor; and a memory connected in communication with the at least one processor; wherein, the memory stores Instructions that can be executed by the at least one processor, the instructions are executed by the at least one processor, so that the at least one processor can execute the method for real-time detection of domestic waste based on the attention mechanism combination of Embodiment 1 .
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