Research on the application of loss function in wood surface defect recognition algorithm
China is a big country in the production, manufacturing and export of wood and wood products, and the surface defects of wood will reduce the external quality and internal strength of wood, and then affect the processing and production process of wood. Based on the self-made wood surface defect data set, this paper conducts training, testing and comparison for the current mainstream detection algorithm models (SSD-300, Faster-Rcnn, YOLOv3, YOLOv4-csp and YOLOv5s). Then, on the basis of YOLOv5s model, 10 kinds of popular loss functions are tested and analyzed comprehensively. The purpose of this paper is to summarize the existing loss functions and expand the application of deep learning model in the field of wood defect detection, in order to provide certain guidance and reference for the automatic detection of wood surface defects.
Authors
- Huan Tan (ORCID: https://orcid.org/0000-0002-2451-9777)
- Qi Wang (ORCID: https://orcid.org/0000-0003-3418-8809)
- Chenghao Lv (ORCID: https://orcid.org/0000-0002-1165-2720)
- Ying Guan (ORCID: https://orcid.org/0000-0003-2320-8325)
- P. Ma
Institutions
- Jiangnan University (CN)
- Nanjing University of Finance and Economics (CN)
- Jiangsu University of Science and Technology (CN)
- Heilongjiang Academy of Agricultural Machinery Engineering (CN)
Publication Details
- Published
- 2025-07-16
- DOI
- https://doi.org/10.1117/12.3070656
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00