Improved U2Net-Based Surface Defect Detection Method for Blister Tablets
Open Access
- 26 September 2024
- journal article
- research article
- Published by MDPI AG in Algorithms
- Vol. 17 (10) , 429
Abstract
Aiming at the problem that the surface defects of blAister tablets are difficult to detect correctly, this paper proposes a detection method based on the improved U2Net. First, the features extracted from the RSU module of U2Net are enhanced and adjusted using the large kernel attention mechanism, so that the U2Net model strengthens its ability to extract defective features. Second, a loss function combining the Gaussian Laplace operator and the cross-entropy function is designed to make the model strengthen its ability to detect edge defects on the surface of blister tablets. Finally, thresholds are adaptively determined using the local mean and OTSU(an adaptive threshold segmentation method) method to improve accuracy. The experimental results show that the method proposed in this paper can reach an average accuracy of 99% and an average precision rate of 96.3%; the model test only takes 50 ms per image, which can meet the rapid detection requirements. Minor surface defects can also be accurately detected, which is better than other algorithmic models of the same type, proving the effectiveness of this method.Keywords
This publication has 9 references indexed in Scilit:
- Fast Gaussian Filter Approximations Comparison on SIMD Computing PlatformsApplied Sciences, 2024
- Visual attention networkComputational Visual Media, 2023
- Pixel-level pruning deep supervision UNet++ for detecting fabric defectsTextile Research Journal, 2023
- Coronary angiography image segmentation based on PSPNetComputer Methods and Programs in Biomedicine, 2021
- U2-Net: Going deeper with nested U-structure for salient object detectionPattern Recognition, 2020
- Automatic Early Broken-Rotor-Bar Detection and Classification Using Otsu SegmentationIEEE Access, 2020
- Mobile-Unet: An efficient convolutional neural network for fabric defect detectionTextile Research Journal, 2020
- Image Segmentation By Using Thresholding Techniques For Medical ImagesComputer Science & Engineering: An International Journal, 2016
- Laplacian Operator-Based Edge DetectorsIEEE Transactions on Pattern Analysis and Machine Intelligence, 2007