Lab Environment | Detail |
---|---|
programming language | Python3.8.5 |
operating system | Windows 10 |
deep learning framework | Pytorch 1.8.0 |
GPU | 4x NVIDIA TITIAN V |
Dataset | Training | Test | Validation | Total |
---|---|---|---|---|
Homemade forest fire data set | 1442 | 617 | 617 | 2676 |
Other institutes data set | 600 | 200 | 200 | 1000 |
Training parameters | Detail |
---|---|
Epochs | 100 |
Batch-size | 16 |
Image-size | 6 40 × 640 |
Initial learning rate | 0.01 |
Optimization algorithm | SGD |
Model | P | R | FPS |
---|---|---|---|
YOLOv5s | 0.811 | 0.786 | 59 |
YOLOv5s + CBAM | 0.814 | 0.790 | 60 |
YOLOv5s + SE | 0.810 | 0.787 | 5 8 |
YOLOv5s +ECA | 0.812 | 0.791 | 5 9 |
YOLOv5s + dsCBAM | 0.812 | 0.787 | 62 |
YOLOv5s + dsCBAM +Alpha-IoU | 0.821 | 0.813 | 61 |
YOLOv5s + dsCBAM + SIoU | 0.860 | 0.834 | 60 |
YOLOv5s + dsCBAM+ VariFocal (Ours) | 0.871 | 0.816 | 64 |