Table 3 Initial parameters of the model.
From: YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithm
Parameter | Value | Note |
|---|---|---|
Training epochs | 300 | Â |
Learning rate | 0.01 | Initial learning rate |
Input size pixels | 640 × 640 |  |
Weight decay | 0.0005 | Â |
Momentum | 0.8 | Â |
Classes loss coefficient | 0.5 | Â |
Box loss coefficient | 0.05 | Â |
Scale (Shape_IOU) | 1.0 | Â |
Validation conf_thres | 0.01 | Â |
Validation iou_thres | 0.6 | Â |
