Development and training of a neural network model for semantic segmentation in terahertz non-destructive testing
摘要
This study presents a neural network model for automatic defect detection and localization in terahertz images of manufactured components. The model employs an architecture based on MBConv blocks and achieves mean IoU values of 0.93 and 0.72 for two defect types, respectively. To eliminate the class imbalance, a 15-fold cross-validation with a weighted loss of the cross entropy is used. Training was performed using the AdamW optimizer with OneCycleLR scheduling and BF16 precision. The results obtained demonstrate that the model can accurately segment defects of various types and sizes, making it suitable for practical non-destructive testing applications.