Table 2 Hyperparameters settings
From: A method for small-sized wheat seedlings detection: from annotation mode to model construction
Model | Batch size | Epoch | Learning rate | Weight decay | Momentum |
|---|---|---|---|---|---|
YOLOv5 | 8 | 300 | 0.010 | 0.0005 | 0.9 |
YOLOv7 | 8 | 300 | 0.010 | 0.0005 | 0.9 |
YOLOv3 | 8 | 300 | 0.010 | 0.0005 | 0.9 |
SSD | 8 | 300 | 0.008 | 0.0005 | 0.9 |
RetinaNet | 8 | 300 | 0.012 | 0.0005 | 0.9 |
Faster-RCNN | 8 | 300 | 0.009 | 0.0005 | 0.9 |