Skip to content

Luckycat518/Yolo-MSAPF

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

39 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Yolo-MSAPF:Weld Surface Defect DETection📌

GitHub Computer Vision in Action License

Table of Contents

Introduction

Up to now, many advanced deep-learning networks have been developed for the detection of weld surface defects. However, the detection speed and accuracy are still limited in the actual industrial environment because the weld surface defects are small and of many types. A great difficulty in location and classification will generate from these diverse small defects. In this work, to further promote the industrial application of the weld surface defect detection network, we aim to solve the industrial problems encountered in weld surface defect detection by providing specific improvement strategies. The fast-response Yolo-v5 model was selected as the baseline network to meet the real-time demand. Considering the existence of a huge amount of interference information on the weld surface, we designed a novel fusion strategy named multi-scale alignment fusion with parallel feature filtering (MSAPF) to integrate and filter multi-scale features. This strategy is implemented by multi-scale alignment fusion (MSAF) and parallel feature filtering (PFF) modules. The MSAF module aims to improve the problems of feature misalignment and spatial feature mismatch during multi-scale feature fusion by aligning features at one level to fuse all other scales. Thus, the sufficiency and accuracy of features are ensured. The subsequent PFF module simultaneously weights the channel and spatial features through the parallel dual-channel attention mechanism to efficiently screen the necessary features and filter out the interference features. Besides, other strategies were also applied to enhance the filtration performance and make lightweight of the whole model. Moreover, a dataset called WSD-DET, which covers the common defect categories occurring in the welding process, was built for weld surface defect detection.

1. Key Issues in Weld Surface Defect Detection

1)External impurity interference

Impurities such as welding slag on the surface of the weld will interfere with the judgment of the edge of the weld and the edge of the defect.

👆 BACK to Table of Contents -->

2)Adjacent interaction interference

It is highly possible to form a variety of defects with different scales on a weld, and these defects will overlap and interfere with each other when identifying their location and classification.

👆 BACK to Table of Contents -->

3)Similarity interference

The optical characteristics of some surface defects are similar. The types of these defects will be confused with each other when the contrast of optical features is not obvious.

👆 BACK to Table of Contents -->

2. Our dataset for Industrial Weld Surface Defect Detection

1)Weld Surface defect: WSD-DET

WSD-DET can be used for classification and positioning tasks.

The WSD-DET contains 7580 pictures with 8 types of defects. The defects for the original unpolished welds include weld_penetration, weld_beading,microporous, hole, incompletely_filled_groove, undercut, crazing and splash. The weld after grinding contains four kinds of defects: hole, undercut, crazing and weld_beading.

👆 BACK to Table of Contents -->

Notification

The first version of the dataset containing the original images appearing in the paper has been released.

The WSD-DET is collected from a commercial company. Therefore, this dataset is limited to academic use!!!Prohibited for any commercial use!!! You can only use this dataset for research purpose. Before the entire dataset is fully released, if you need this dataset to do some research, please contact us. After review and permission by the commercial company, we will provide the dataset to you.

If you have any questions or idea, please let me know

(email: wgq001@csu.edu.cn; 2013801003@csu.edu.cn)

👆 BACK to Table of Contents -->

Citation

This paper is still under review......

👆 BACK to Table of Contents -->

License

Distributed under the MIT License. See LICENSE for more information.

About

Yolo-MSAPF: Multi-Scale Alignment fusion with Parallel feature Filtering model for high accuracy weld defect detection

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published