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