Defective workpiece image recognition method based on convolutional attention neural network

A convolutional neural network and neural network technology, applied in the field of defect workpiece image recognition based on convolutional attention neural network, can solve the problems of small defects being easily ignored and low recognition rate, so as to improve the accuracy rate and expand the data set , a wide range of effects

Pending Publication Date: 2020-08-04
暖屋信息科技(苏州)有限公司
View PDF5 Cites 8 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

[0004] Aiming at the defects of the above-mentioned prior art, the present invention provides a defect workpiece image recognition method based on a convolutional attention neural network, which solves the problem of low recognition rate caused by small defects being easily overlooked

Method used

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
View more

Image

Smart Image Click on the blue labels to locate them in the text.
Viewing Examples
Smart Image
  • Defective workpiece image recognition method based on convolutional attention neural network
  • Defective workpiece image recognition method based on convolutional attention neural network

Examples

Experimental program
Comparison scheme
Effect test

Embodiment Construction

[0021] The present invention will be further described below in conjunction with embodiment, should be understood that these embodiments are only used to illustrate the present invention and are not intended to limit the scope of the present invention, after having read the present invention, those skilled in the art can modify various equivalent forms of the present invention All fall within the scope defined by the appended claims of this application.

[0022] Please combine figure 1 As shown, the present embodiment relates to a defect workpiece image recognition method based on a convolutional attention neural network, including

[0023] Feature extraction: Based on the network structure of the convolutional neural network, the feature extraction network is formed after pre-training the weights on the dataset ImageNet, and the image X∈R (H×W) , input into the feature extraction network to obtain the corresponding deep features;

[0024] The network structure of the convol...

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

PUM

No PUM Login to view more

Abstract

The invention discloses a defective workpiece image recognition method based on a convolutional attention neural network. The method comprises the steps of performing pre-training weight on the convolutional neural network to form a deep feature for input image input, of a feature extraction network; carrying out convolution operation on the depth feature image to obtain an attention graph, and performing normalization; carrying out attention region cutting and discarding on the normalized attention map, and inputting the attention region cutting and discarding into the convolutional neural network again for training; inputting a defective workpiece image to be recognized into the trained convolutional neural network to obtain a feature map, and performing dot product operation on the feature map and the attention map to obtain a new part feature map; performing maximum pooling operation on the part feature map to obtain attention features, stacking all the attention features to obtaina part feature matrix, and completing classification and recognition of defect images. According to the invention, the accuracy of workpiece defect detection is improved, and the method can be applied to various tiny defect detection tasks.

Description

technical field [0001] The invention relates to a defect workpiece image recognition method, in particular to a defect workpiece image recognition method based on a convolutional attention neural network. Background technique [0002] There may be a variety of defects in workpiece casting production. The defects on the surface of the workpiece can be judged by direct observation, but some defects exist inside the workpiece, which cannot be judged by observation. X-ray images are often used for non-destructive testing. The discrimination of X-ray images has been done manually in the past, and the discrimination efficiency is low. With the development of artificial intelligence, especially the development of algorithms such as machine learning, the application of computers for defect discrimination has improved efficiency. However, the machine learning algorithm also has the problem of poor versatility. It is necessary to build a corresponding model for a workpiece, and the s...

Claims

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

Application Information

Patent Timeline
no application Login to view more
IPC IPC(8): G06T7/00
CPCG06T7/0004G06T2207/20081G06T2207/20084G06T2207/30164Y02P90/30
Inventor 王永雄蒋莉莉刘智华
Owner 暖屋信息科技(苏州)有限公司
Who we serve
  • R&D Engineer
  • R&D Manager
  • IP Professional
Why Eureka
  • Industry Leading Data Capabilities
  • Powerful AI technology
  • Patent DNA Extraction
Social media
Try Eureka
PatSnap group products

Browse by: Latest US Patents, China's latest patents, Technical Efficacy Thesaurus, Application Domain, Technology Topic.

© 2024 PatSnap. All rights reserved.Legal|Privacy policy|Modern Slavery Act Transparency Statement|Sitemap