CN114170140A - Membrane defect identification method based on Yolov4 - Google Patents
Membrane defect identification method based on Yolov4 Download PDFInfo
- Publication number
- CN114170140A CN114170140A CN202111325709.5A CN202111325709A CN114170140A CN 114170140 A CN114170140 A CN 114170140A CN 202111325709 A CN202111325709 A CN 202111325709A CN 114170140 A CN114170140 A CN 114170140A
- Authority
- CN
- China
- Prior art keywords
- diaphragm
- network model
- module
- training
- detection
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 230000007547 defect Effects 0.000 title claims abstract description 72
- 238000000034 method Methods 0.000 title claims abstract description 29
- 239000012528 membrane Substances 0.000 title description 5
- 238000001514 detection method Methods 0.000 claims abstract description 46
- 238000012549 training Methods 0.000 claims abstract description 31
- 238000012360 testing method Methods 0.000 claims abstract description 15
- 230000007246 mechanism Effects 0.000 claims abstract description 9
- 208000034564 Coronary ostial stenosis or atresia Diseases 0.000 claims description 12
- 238000000605 extraction Methods 0.000 claims description 10
- 230000002950 deficient Effects 0.000 claims description 9
- 238000002372 labelling Methods 0.000 claims description 9
- 230000004927 fusion Effects 0.000 claims description 8
- 238000005070 sampling Methods 0.000 claims description 8
- 238000012216 screening Methods 0.000 claims description 7
- 238000012795 verification Methods 0.000 claims description 6
- 230000005764 inhibitory process Effects 0.000 claims description 3
- 239000000284 extract Substances 0.000 claims description 2
- 230000006870 function Effects 0.000 description 7
- 230000000694 effects Effects 0.000 description 6
- 238000004519 manufacturing process Methods 0.000 description 6
- WHXSMMKQMYFTQS-UHFFFAOYSA-N Lithium Chemical compound [Li] WHXSMMKQMYFTQS-UHFFFAOYSA-N 0.000 description 5
- 229910052744 lithium Inorganic materials 0.000 description 5
- 238000012545 processing Methods 0.000 description 5
- 238000004364 calculation method Methods 0.000 description 4
- 238000010586 diagram Methods 0.000 description 4
- 238000005507 spraying Methods 0.000 description 4
- 230000004913 activation Effects 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 2
- 230000015654 memory Effects 0.000 description 2
- 238000012986 modification Methods 0.000 description 2
- 230000004048 modification Effects 0.000 description 2
- 238000006467 substitution reaction Methods 0.000 description 2
- 238000013528 artificial neural network Methods 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 238000005520 cutting process Methods 0.000 description 1
- 238000013461 design Methods 0.000 description 1
- 238000009826 distribution Methods 0.000 description 1
- 238000003709 image segmentation Methods 0.000 description 1
- 230000000877 morphologic effect Effects 0.000 description 1
- 230000008569 process Effects 0.000 description 1
- 239000002994 raw material Substances 0.000 description 1
- 238000005096 rolling process Methods 0.000 description 1
- 238000011179 visual inspection Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/0002—Inspection of images, e.g. flaw detection
- G06T7/0004—Industrial image inspection
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/23—Clustering techniques
- G06F18/232—Non-hierarchical techniques
- G06F18/2321—Non-hierarchical techniques using statistics or function optimisation, e.g. modelling of probability density functions
- G06F18/23213—Non-hierarchical techniques using statistics or function optimisation, e.g. modelling of probability density functions with fixed number of clusters, e.g. K-means clustering
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/25—Fusion techniques
- G06F18/253—Fusion techniques of extracted features
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- General Physics & Mathematics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Computing Systems (AREA)
- Biomedical Technology (AREA)
- General Health & Medical Sciences (AREA)
- Computational Linguistics (AREA)
- Biophysics (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Molecular Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Health & Medical Sciences (AREA)
- Evolutionary Biology (AREA)
- Probability & Statistics with Applications (AREA)
- Quality & Reliability (AREA)
- Image Analysis (AREA)
Abstract
The invention provides a method for identifying diaphragm defects based on Yolov4, which comprises the following steps: firstly, acquiring diaphragm image data; secondly, marking the defect type by using a marking tool to generate a marking file as a data set; thirdly, clustering a prior frame of the data set by using a clustering algorithm k-means; fourthly, establishing an improved Yolov4 network model, and training the improved Yolov4 network model through a training set; the improved Yolov4 network model adopts a structure of combining low-level characteristic information and high-level characteristic information; CSP structures are added into SPP and PANet of the improved Yolov4 network model, and an attention mechanism is combined to improve the defect detection precision; fifthly, testing the detection performance of the trained improved Yolov4 network model by using a test set; and sixthly, deploying the training model of the optimal improved Yolov4 network model to a diaphragm detection site for diaphragm defect detection. The method has strong robustness, can reduce the omission factor and the false detection rate, and can improve the detection quality of the diaphragm.
Description
Technical Field
The invention relates to the technical field of diaphragm detection, in particular to a method for identifying diaphragm defects based on Yolov 4.
Background
The lithium battery is visible everywhere in daily life, such as a mobile phone, a battery car, a new energy automobile and the like which are commonly used. The demand is increasing, and lithium cell industry is also continuous to put into production, but some problems are also emerging gradually, for example, a certain brand cell-phone charges and explodes, or the battery has the problem such as electric leakage. The safety and durability of the battery need to be further enhanced. The diaphragm is an important component of the lithium battery, isolates the positive electrode and the negative electrode of the battery and provides a flow passage for the battery.
However, in an automatic production line, the membrane is inevitably subjected to some collisions, folding and friction, and the surface of the membrane may have creases, scratches and pinholes, or factors such as missing spraying or less spraying in the spraying process of the membrane. As a raw material of the battery, the quality of the separator also affects the quality and safety of the battery, and therefore, it is necessary to check the production quality of the separator.
At the present stage, the manual diaphragm detection mode is gradually eliminated, and the detection efficiency cannot meet the production requirements of modern industry. With the continuous progress of cameras and image algorithms, visual inspection is increasingly applied to the industry. Since the diaphragm is produced at first and is subjected to rolling and splitting procedures, the diaphragm is conveyed and photographed on a conveyor belt for detection, and some manufacturers choose to use a line-scan camera to acquire an image of the diaphragm and use traditional image algorithms such as morphological processing, image segmentation and the like to detect the quality of the diaphragm. However, in an automatic production mode, the conventional image processing algorithm has a considerable effect on detection of defect features, but the defects on the diaphragm have different forms, and gray values of some defects are similar to the background of a normal diaphragm, so that the conventional image processing algorithm has a high probability of missing detection and false detection in this respect, and meanwhile, the scheme design of the conventional image processing algorithm is also largely related to the technical level of technicians, and the production environment also affects the detection effect.
Disclosure of Invention
The invention aims to overcome the defects and shortcomings in the prior art, and provides a method for identifying diaphragm defects based on Yolov4, which has strong robustness, can reduce the omission factor and the false detection factor, and can solve the problem that the traditional algorithm cannot detect the fine defects of the diaphragm, thereby achieving better positioning and classifying effects on the diaphragm defects and further improving the detection quality of the diaphragm.
In order to achieve the purpose, the invention is realized by the following technical scheme: a method for identifying diaphragm defects based on Yolov4 is characterized in that: the method comprises the following steps:
firstly, acquiring diaphragm image data, and screening defective pictures from the image data;
secondly, marking the defect type by using a marking tool to generate a marking file as a data set;
thirdly, clustering a prior frame of the data set by using a clustering algorithm k-means;
fourthly, dividing the data set into a training set, a verification set and a test set; establishing an improved Yolov4 network model, and training the improved Yolov4 network model through a training set; the improved Yolov4 network model adopts a structure of combining low-level characteristic information and high-level characteristic information; CSP structures are added into SPP and PANet of the improved Yolov4 network model, and an attention mechanism is combined to improve the defect detection precision;
fifthly, testing the detection performance of the trained improved Yolov4 network model by using the test set, and using the optimal training model of the improved Yolov4 network model for detection;
and sixthly, deploying the training model of the optimal improved Yolov4 network model to a diaphragm detection site for diaphragm defect detection, decoding the prediction result, performing score sorting and non-maximum inhibition screening on the prediction result, and drawing the processed result on the diaphragm original image.
In the first step, after defective pictures are screened from image data, the defective pictures are cut randomly; and rotating and folding the cut image to increase sample data.
Clockwise 90 DEG, 180 DEG and 270 DEG rotation is carried out on the cut image; and horizontally folding and vertically folding the cut image. The method can enhance the sample data to expand the data set, avoid overfitting, simultaneously improve the generalization ability of the subsequent model, and enhance the diversity of the sample by using the data.
In the second step, the defect type is marked by using a marking tool, and a marking file is generated as a data set, wherein the marking file is as follows: and manually labeling the image with the diaphragm defects by using a labelImg picture labeling tool, labeling the types of the defects and the minimum external rectangular frame of the defects, and generating an xml file format for storing the sizes and the types of the pictures of the defects and the coordinate information of the labeled rectangular frame as a data set.
The improved Yolov4 network model comprises a module 1, a module 2, a module 3 and a Head network Yolo Head; the module 1 is a trunk feature extraction network COSA-2x2 x; the module 2 adds a CSP structure on the basis of SPP; the module 3 adds a CSP structure to the up-sampling and down-sampling of the PANET structure and adds an attention mechanism.
And adding a feature fusion layer on the top feature layer of the module 3, wherein the feature fusion layer comprises 3 convolution layers, and the combination of low-layer feature information and high-layer feature information is realized.
The CSP structure firstly divides an input part into two parts, then one part passes through a Conv block part of a main body, and the other part is directly stacked with an output after the Conv block; wherein, dividing the input part into two parts means that the input part is divided into two parts on average, and the number of channels is half of the original input; or dividing the input part into two means performing convolution by 1 × 1, and halving the number of channels.
The fourth step includes the steps of:
dividing a data set into a training set, a verification set and a test set according to the proportion of 0.7:0.15: 0.15;
step two, improving a Yolov4 network model for initialization;
thirdly, obtaining an output value of the input training set data through a backbone feature extraction network COSA-2x2x, a module 2, a module 3 and a Head network Yolo Head;
step four, solving the error between the output value of the improved Yolov4 network model and the target value, namely a loss function;
and step five, updating the weight, and finishing the training when the improved Yolov4 network model converges to a certain degree and does not drop any more.
In the third step, the training set data extracts the defect characteristic information of the diaphragm through a trunk characteristic extraction network COSA-2x2x, the receptive field is increased through a module 2, the context information is separated, the characteristics are repeatedly extracted through a module 3, and finally the obtained characteristics are predicted through a Head network Yolo Head.
The loss function in step four is
loss＝lossBoundary frame+lossConfidence level+lossClassification；
lossBoundary frame＝lossCDIoU＝1-IoU+ρ2(box_dt,box_gt)/c2+αν；
The CSPdacrnet 53 has strong feature extraction capability without using the original trunk network CSPdacrnet 53 and CSPdacrnet 53 of yolov4, but a large amount of display card memories occupied by calculation and network training are correspondingly brought, and the detection speed is slower than that of COSA-2x2 x. The COSA-2x2x with the PCB technology can make the model more flexible, and the speed is greatly improved. In addition, the CSP structure added on the basis of the SPP can effectively reduce the calculation amount of the model under the condition of little precision loss.
CSP structures are added to the up-sampling and down-sampling of the PANet structure, and an attention mechanism is integrated into the PANet structure, so that an improved Yolov4 network model can pay more attention to an ROI (region of interest) in a defect, the key information of the defect of the diaphragm is extracted, most irrelevant background information of the diaphragm of the lithium battery is ignored, the number of the parameters can be reduced, and the detection precision is improved. Because the diaphragm has some tiny defects, the invention adds a feature fusion layer on the top feature layer of the module 3 to realize the combination of low-layer feature information and high-layer feature information and carry out subsequent defect detection and positioning, so that the output feature layers are (104, 104 and 128), the micro-miniature defect detection device is more suitable for detecting micro-miniature targets, and has higher positioning precision and reduced false detection rate of tiny defects.
Compared with the prior art, the invention has the following advantages and beneficial effects: the method for identifying the diaphragm defects based on the Yolov4 has strong robustness, can reduce the omission factor and the false detection rate, and can solve the problem that the traditional algorithm cannot detect the fine defects of the diaphragm, thereby achieving better positioning and classifying effects on the diaphragm defects and further improving the detection quality of the diaphragm.
Drawings
FIG. 1 is a flow chart of a Yolov 4-based method for identifying diaphragm defects in accordance with the present invention;
FIG. 2 is a schematic diagram of the improved Yolov4 network model of the present invention;
FIG. 3 is a CSP structure diagram in the improved Yolov4 network model according to the present invention;
FIG. 4 is a diagram of CSP structure and 5 convolutions in the improved Yolov4 network model according to the present invention;
5(a) -5(e) are graphs showing the detection effect of the method for identifying the defect of the diaphragm based on Yolov 4;
Detailed Description
The present invention will be described in further detail with reference to the accompanying drawings and specific embodiments.
Examples
As shown in fig. 1 to 5(e), the method for identifying a defect of a diaphragm based on Yolov4 of the present invention includes the following steps:
firstly, acquiring diaphragm image data, and screening defective pictures from the image data; after screening defective pictures from the image data, randomly cutting the defective pictures; and rotating the cut image by 90 degrees, 180 degrees and 270 degrees clockwise, and performing horizontal folding and vertical folding to increase sample data.
And secondly, manually labeling the image with the diaphragm defects by using a labelImg picture labeling tool, labeling the types of the defects and the minimum external rectangular frame of the defects, and generating an xml file format for storing the sizes and the types of the pictures of the defects and the coordinate information of the labeled rectangular frame as a data set.
And thirdly, clustering 9 groups of prior frames of the data set by using a clustering algorithm k-means, and taking the 9 groups of prior frames as an anchor box of the improved Yolov4 network model in the fourth step, so that the prior frames are more in line with the size and distribution condition of the rectangular frame of the diaphragm defect.
Fourthly, dividing the data set into a training set, a verification set and a test set; establishing an improved Yolov4 network model, and training the improved Yolov4 network model through a training set; the improved Yolov4 network model adopts a structure of combining low-level characteristic information and high-level characteristic information; CSP structures are added into SPP and PANet of the improved Yolov4 network model, and an attention mechanism is combined to improve the defect detection precision;
fifthly, testing the detection performance of the trained improved Yolov4 network model by using the test set, and using the optimal training model of the improved Yolov4 network model for detection;
sixthly, deploying a training model of an optimal improved Yolov4 network model to a diaphragm detection site to perform diaphragm defect detection, decoding a prediction result, performing score sorting and non-maximum inhibition screening on the prediction result, and drawing a processing result on a diaphragm original image, wherein fig. 5(a) -5(e) are effect images for respectively detecting creases, black spots, missed spraying, scratches and pinholes by adopting the Yolov 4-based diaphragm defect identification method.
Specifically, the improved Yolov4 network model comprises a module 1, a module 2, a module 3 and a Head network Yolo Head, wherein the module 1 is a backbone feature extraction network COSA-2x2x, the module 2 is a CSP structure added on the basis of SPP, and the module 3 is a CSP structure added on the upsampling and downsampling of a PANet structure and an attention mechanism is added. According to the invention, a feature fusion layer is added on the top feature layer of the module 3, and the feature fusion layer comprises 3 convolution layers, so that the combination of low-layer feature information and high-layer feature information is realized.
As shown in fig. 3, the CSP structure first divides the input part into two parts, then one part passes through the trunk Conv block part, and the other part is directly stacked with the output after passing through the Conv block; wherein, dividing the input part into two parts means that the input part is divided into two parts on average, and the number of channels is half of the original input; or dividing the input part into two means performing convolution by 1 × 1, and halving the number of channels.
The CSPdacrnet 53 has strong feature extraction capability without using the original trunk network CSPdacrnet 53 and CSPdacrnet 53 of yolov4, but a large amount of display card memories occupied by calculation and network training are correspondingly brought, and the detection speed is slower than that of COSA-2x2 x. The COSA-2x2x with the PCB technology can make the model more flexible, and the speed is greatly improved. In addition, the CSP structure added on the basis of the SPP can effectively reduce the calculation amount of the model under the condition of little precision loss.
The CSP structure is added to the up-sampling and the down-sampling of the PANet structure, the attention mechanism is integrated into the PANet structure, an improved Yolov4 network model is enabled to pay more attention to an ROI (region of interest) in a defect, key information of the defect of the diaphragm is extracted, most irrelevant background information of the diaphragm of the lithium battery is ignored, the number of the parameters can be reduced, and the detection precision can be improved, wherein the CSP structure and a 5-time convolution schematic diagram are shown in figure 4. Because the diaphragm has some tiny defects, the invention adds a feature fusion layer on the top feature layer of the module 3 to realize the combination of low-layer feature information and high-layer feature information and carry out subsequent defect detection and positioning, so that the output feature layers are (104, 104 and 128), the micro-miniature defect detection device is more suitable for detecting micro-miniature targets, and has higher positioning precision and reduced false detection rate of tiny defects.
The Yolo Head of the present invention uses the extracted features to perform prediction, and predicts 3 feature scales including width, height and channel number, which are (104, 104, 128), (26, 26, 512), (13, 13, 1024), respectively. All convolutions with a convolution kernel of 1x1 of the present invention use the Mish activation function. Mish activation function has better smoothness than Leaky-ReLU, and can allow better information to be transmitted into a neural network, so that better accuracy and generalization capability are obtained. The expression of the Mish activation function is as follows:
Mish＝x×tanh(In(1+ex))
the fourth step of the present invention comprises the steps of:
step one, dividing a data set into a training set, a verification set and a test set according to the proportion of 0.7:0.15: 0.15.
And step two, initializing the improved Yolov4 network model.
And step three, extracting defect characteristic information of the diaphragm by the training set data through a trunk characteristic extraction network COSA-2x2x, increasing the receptive field through a module 2, separating context information, repeatedly extracting characteristics through a module 3, and finally predicting the obtained characteristics through a Head network Yolo Head.
Step four, solving the error between the output value of the improved Yolov4 network model and the target value, namely a loss function; wherein the loss function is
loss＝lossBoundary frame+lossConfidence level+lossClassification；
lossBoundary frame＝lossCDIoU＝1-IoU+ρ2(box_dt,box_gt)/c2+αν；
And step five, updating the weight, and finishing the training when the improved Yolov4 network model converges to a certain degree and does not drop any more.
The detection performance of the Yolov 4-based diaphragm defect identification method adopted in the present embodiment is shown in table 1 below.
TABLE 1 test performance of the model
| Speed of detection (ms/piece) | Recall (%) | Model parameter quantity params | |
| Original model algorithm | 54 | 88.13 | 64,040,001 |
| The example method | 28 | 91.80 | 15,178,913 |
| SSD | 20 | 48.18 | 6,204,004 |
| YOLOv4-tiny | 9 | 75.35 | 5,961,014 |
Wherein, the recall ratio formula is as follows:
the True Positives (TP) indicates that it is predicted to be a positive sample, actually a positive sample. False Positive (FP) indicates a predicted positive sample, and actually a negative sample. The higher the recall, the higher the probability that the actual defect is predicted.
The above embodiments are preferred embodiments of the present invention, but the present invention is not limited to the above embodiments, and any other changes, modifications, substitutions, combinations, and simplifications which do not depart from the spirit and principle of the present invention should be construed as equivalents thereof, and all such changes, modifications, substitutions, combinations, and simplifications are intended to be included in the scope of the present invention.
Claims (10)
1. A method for identifying diaphragm defects based on Yolov4 is characterized in that: the method comprises the following steps:
firstly, acquiring diaphragm image data, and screening defective pictures from the image data;
secondly, marking the defect type by using a marking tool to generate a marking file as a data set;
thirdly, clustering a prior frame of the data set by using a clustering algorithm k-means;
fourthly, dividing the data set into a training set, a verification set and a test set; establishing an improved Yolov4 network model, and training the improved Yolov4 network model through a training set; the improved Yolov4 network model adopts a structure of combining low-level characteristic information and high-level characteristic information; CSP structures are added into SPP and PANet of the improved Yolov4 network model, and an attention mechanism is combined to improve the defect detection precision;
fifthly, testing the detection performance of the trained improved Yolov4 network model by using the test set, and using the optimal training model of the improved Yolov4 network model for detection;
and sixthly, deploying the training model of the optimal improved Yolov4 network model to a diaphragm detection site for diaphragm defect detection, decoding the prediction result, performing score sorting and non-maximum inhibition screening on the prediction result, and drawing the processed result on the diaphragm original image.
2. The Yolov 4-based diaphragm defect identification method according to claim 1, wherein: in the first step, after defective pictures are screened from image data, the defective pictures are cut randomly; and rotating and folding the cut image to increase sample data.
3. The Yolov 4-based diaphragm defect identification method according to claim 2, wherein: clockwise 90 DEG, 180 DEG and 270 DEG rotation is carried out on the cut image; and horizontally folding and vertically folding the cut image.
4. The Yolov 4-based diaphragm defect identification method according to claim 1, wherein: in the second step, the defect type is marked by using a marking tool, and a marking file is generated as a data set, wherein the marking file is as follows: and manually labeling the image with the diaphragm defects by using a labelImg picture labeling tool, labeling the types of the defects and the minimum external rectangular frame of the defects, and generating an xml file format for storing the sizes and the types of the pictures of the defects and the coordinate information of the labeled rectangular frame as a data set.
5. The Yolov 4-based diaphragm defect identification method according to claim 1, wherein: the improved Yolov4 network model comprises a module 1, a module 2, a module 3 and a Head network Yolo Head; the module 1 is a trunk feature extraction network COSA-2x2 x; the module 2 adds a CSP structure on the basis of SPP; the module 3 adds a CSP structure to the up-sampling and down-sampling of the PANET structure and adds an attention mechanism.
6. The Yolov 4-based diaphragm defect identification method according to claim 5, wherein: and adding a feature fusion layer on the top feature layer of the module 3, wherein the feature fusion layer comprises 3 convolution layers, and the combination of low-layer feature information and high-layer feature information is realized.
7. The Yolov 4-based diaphragm defect identification method according to claim 5, wherein: the CSP structure firstly divides an input part into two parts, then one part passes through a Conv block part of a main body, and the other part is directly stacked with an output after the Conv block; wherein, dividing the input part into two parts means that the input part is divided into two parts on average, and the number of channels is half of the original input; or dividing the input part into two means performing convolution by 1 × 1, and halving the number of channels.
8. The Yolov 4-based diaphragm defect identification method according to claim 6, wherein: the fourth step includes the steps of:
dividing a data set into a training set, a verification set and a test set according to the proportion of 0.7:0.15: 0.15;
step two, improving a Yolov4 network model for initialization;
thirdly, obtaining an output value of the input training set data through a backbone feature extraction network COSA-2x2x, a module 2, a module 3 and a Head network Yolo Head;
step four, solving the error between the output value of the improved Yolov4 network model and the target value, namely a loss function;
and step five, updating the weight, and finishing the training when the improved Yolov4 network model converges to a certain degree and does not drop any more.
9. The Yolov 4-based diaphragm defect identification method according to claim 8, wherein: in the third step, the training set data extracts the defect characteristic information of the diaphragm through a trunk characteristic extraction network COSA-2x2x, the receptive field is increased through a module 2, the context information is separated, the characteristics are repeatedly extracted through a module 3, and finally the obtained characteristics are predicted through a Head network Yolo Head.
10. The Yolov 4-based diaphragm defect identification method according to claim 8, wherein: the loss function in step four is
loss＝lossBoundary frame+lossConfidence level+lossClassification；
lossBoundary frame＝lossCDIoU＝1-IoU+ρ2(box_dt,box_gt)/c2+αν；
Priority Applications (1)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| CN202111325709.5A CN114170140A (en) | 2021-11-10 | 2021-11-10 | Membrane defect identification method based on Yolov4 |
Applications Claiming Priority (1)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| CN202111325709.5A CN114170140A (en) | 2021-11-10 | 2021-11-10 | Membrane defect identification method based on Yolov4 |
Publications (1)
| Publication Number | Publication Date |
|---|---|
| CN114170140A true CN114170140A (en) | 2022-03-11 |
Family
ID=80478538
Family Applications (1)
| Application Number | Title | Priority Date | Filing Date |
|---|---|---|---|
| CN202111325709.5A Pending CN114170140A (en) | 2021-11-10 | 2021-11-10 | Membrane defect identification method based on Yolov4 |
Country Status (1)
| Country | Link |
|---|---|
| CN (1) | CN114170140A (en) |
Cited By (6)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN114723750A (en) * | 2022-06-07 | 2022-07-08 | 南昌大学 | Transmission line strain clamp defect detection method based on improved YOLOX algorithm |
| CN115082401A (en) * | 2022-06-22 | 2022-09-20 | 桂林电子科技大学 | SMT production line chip mounter fault prediction method based on improved YOLOX and PNN |
| CN115100483A (en) * | 2022-03-28 | 2022-09-23 | 中车贵阳车辆有限公司 | Open wagon body defect identification method based on deep learning |
| CN115588022A (en) * | 2022-11-10 | 2023-01-10 | 合肥惠强新能源材料科技有限公司 | Lithium battery isolation film quality detection system based on process index data |
| CN116758064A (en) * | 2023-08-14 | 2023-09-15 | 深圳天眼新能源科技有限公司 | Lithium battery diaphragm quality detection method based on electron scanning microscope |
| WO2023202207A1 (en) * | 2022-04-18 | 2023-10-26 | 宁德时代新能源科技股份有限公司 | Method, device and system for detecting defects of separator for composite material strip |
-
2021
- 2021-11-10 CN CN202111325709.5A patent/CN114170140A/en active Pending
Cited By (8)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN115100483A (en) * | 2022-03-28 | 2022-09-23 | 中车贵阳车辆有限公司 | Open wagon body defect identification method based on deep learning |
| WO2023202207A1 (en) * | 2022-04-18 | 2023-10-26 | 宁德时代新能源科技股份有限公司 | Method, device and system for detecting defects of separator for composite material strip |
| CN114723750A (en) * | 2022-06-07 | 2022-07-08 | 南昌大学 | Transmission line strain clamp defect detection method based on improved YOLOX algorithm |
| CN115082401A (en) * | 2022-06-22 | 2022-09-20 | 桂林电子科技大学 | SMT production line chip mounter fault prediction method based on improved YOLOX and PNN |
| CN115082401B (en) * | 2022-06-22 | 2024-03-19 | 桂林电子科技大学 | SMT production line chip mounter fault prediction method based on improved YOLOX and PNN |
| CN115588022A (en) * | 2022-11-10 | 2023-01-10 | 合肥惠强新能源材料科技有限公司 | Lithium battery isolation film quality detection system based on process index data |
| CN116758064A (en) * | 2023-08-14 | 2023-09-15 | 深圳天眼新能源科技有限公司 | Lithium battery diaphragm quality detection method based on electron scanning microscope |
| CN116758064B (en) * | 2023-08-14 | 2024-04-19 | 深圳天眼新能源科技有限公司 | Lithium battery diaphragm quality detection method based on electron scanning microscope |
Similar Documents
| Publication | Publication Date | Title |
|---|---|---|
| CN114170140A (en) | Membrane defect identification method based on Yolov4 | |
| CN111598861B (en) | Improved Faster R-CNN model-based non-uniform texture small defect detection method | |
| CN111415329B (en) | Workpiece surface defect detection method based on deep learning | |
| CN113724231B (en) | Industrial defect detection method based on semantic segmentation and target detection fusion model | |
| CN106875373B (en) | Mobile phone screen MURA defect detection method based on convolutional neural network pruning algorithm | |
| CN104392205B (en) | A kind of recognition methods and system of improper license plate | |
| CN111695514A (en) | Vehicle detection method in foggy days based on deep learning | |
| CN110544231B (en) | Lithium battery electrode surface defect detection method based on background standardization and centralized compensation algorithm | |
| CN110610475A (en) | Visual defect detection method of deep convolutional neural network | |
| CN109544522A (en) | A kind of Surface Defects in Steel Plate detection method and system | |
| CN112258470B (en) | Intelligent industrial image critical compression rate analysis system and method based on defect detection | |
| CN111291806A (en) | Identification method of label number of industrial product based on convolutional neural network | |
| CN110415238A (en) | Diaphragm spots detection method based on reversed bottleneck structure depth convolutional network | |
| CN114972316A (en) | Battery case end surface defect real-time detection method based on improved YOLOv5 | |
| CN115456955A (en) | Method for detecting internal burr defect of ball cage dust cover | |
| CN111507391A (en) | Intelligent identification method for nonferrous metal broken materials based on machine vision | |
| CN115184364A (en) | Tire bubble defect detection method and system based on 3D point cloud | |
| CN113781483B (en) | Industrial product appearance defect detection method and device | |
| CN110508510A (en) | A kind of plastic pump defect inspection method, apparatus and system | |
| CN116363136B (en) | On-line screening method and system for automatic production of motor vehicle parts | |
| CN112347967B (en) | Pedestrian detection method fusing motion information in complex scene | |
| CN113538342A (en) | Convolutional neural network-based quality detection method for coating of aluminum aerosol can | |
| CN116486228A (en) | Paper medicine box steel seal character recognition method based on improved YOLOV5 model | |
| CN116228687A (en) | Chip package defect detection method and device based on deep learning | |
| CN114113137A (en) | Defect detection system and method for thin film material |
Legal Events
| Date | Code | Title | Description |
|---|---|---|---|
| PB01 | Publication | ||
| PB01 | Publication | ||
| SE01 | Entry into force of request for substantive examination | ||
| SE01 | Entry into force of request for substantive examination | ||
| CB02 | Change of applicant information | ||
| CB02 | Change of applicant information |
Country or region after: China Address after: No.33, Guangyun Road, Nanhai District, Foshan City, Guangdong Province, 528200 Applicant after: Foshan University Address before: No.33, Guangyun Road, Nanhai District, Foshan City, Guangdong Province, 528200 Applicant before: FOSHAN University Country or region before: China |