Table 4 Parameter settings of the six computer vision models.
From: A Swin Transformer-based model for mosquito species identification
Model | Numbers of epochs | Learning rate | Batch-size |
|---|---|---|---|
Detection Transformer | 300 | 0.0001 | 2 |
Vision Transformer | 500 | 0.03 | 512 |
Swin Transformer | 300 | 0.0005 | 128 |
YOLOv5 | 300 | 0.001 | 16 |
Mask R-CNN | 160 | 0.001 | 1 |
DenseNet | 300 | 0.1 | 16 |
