Yi WANG, Xiaojie GONG, Hao SU. Image segmentation method of surface defects for metal workpieces based on improved U-net[J]. Journal of Applied Optics, 2023, 44(1): 86
- Journal of Applied Optics
- Vol. 44, Issue 1, 86 (2023)
Fig. 1. Structure diagram of U-net network
Fig. 2. Structure diagram of CBAM network
Fig. 3. Structure diagram of improved U-net network
Fig. 4. Visual platform for image acquisition
Fig. 5. Comparison before and after image preprocessing
Fig. 6. Variation curves of loss values
Fig. 7. Variation curves of mean intersection over union
Fig. 8. Variation curves of accuracy rate
Fig. 9. Comparison of segmentation results
|
Table 1. Comparison of Dice indexes
Set citation alerts for the article
Please enter your email address
© Copyright 2018-2021 | Chinese Laser Press. All Rights Reserved 沪ICP备15018463号-20