Official websites use .gov
A
.gov website belongs to an official
government organization in the United States.
Secure .gov websites use HTTPS
A lock (
) or https:// means you've safely
connected to the .gov website. Share sensitive
information only on official, secure websites.
Training parameters used in network.
| Training Parameters | YOLOv5n | YOLOv5-CS |
|---|---|---|
| Initial learning rate | 0.01 | 0.01 |
| learning rate | 0.0001 | 0.0001 |
| Batch size | 12 | 12 |
| Epochs | 200 | 200 |
| Train crop size | 640 × 640 | 640 × 640 |
| Class | 1 | 1 |
| IoU Threshold | 0.5 | 0.5 |
| Confidence Threshold | 0.25 | 0.25 |
