robmarkcole / fire-detection-from-images

Detect fire in images using neural nets
MIT License
376 stars 76 forks source link

Try new yolos #27

Closed robmarkcole closed 2 years ago

robmarkcole commented 2 years ago
robmarkcole commented 2 years ago

YoloR gives great result without even augmentation:

 P           R      mAP@.5      mAP@.5:.95
0.681     0.833       0.831          0.408
robmarkcole commented 2 years ago

Scaled yolo-v4

P               R     mAP@.5      mAP@.5:.95:
0.443       0.775       0.705          0.281
robmarkcole commented 2 years ago

Yolov4-tiny

precision = 0.69, recall = 0.59, F1-score = 0.64 

mAP@0.50 = 0.663356
robmarkcole commented 2 years ago

done