Table 1 Training parameters.
From: Metal surface defect detection based on improved YOLOv5
Parameter | Value |
|---|---|
Learning rate | 0.01 |
Batch size | 8 |
Image size | 640*640 |
Weight decay | 0.0005 |
Momentum | 0.937 |
Epoch | 300 |
From: Metal surface defect detection based on improved YOLOv5
Parameter | Value |
|---|---|
Learning rate | 0.01 |
Batch size | 8 |
Image size | 640*640 |
Weight decay | 0.0005 |
Momentum | 0.937 |
Epoch | 300 |