robmarkcole / fire-detection-from-images

Detect fire in images using neural nets
MIT License
379 stars 75 forks source link

Compare yolov5 with retinanet #11

Open robmarkcole opened 3 years ago

robmarkcole commented 3 years ago

cocometric reported by retinanet (icevision) is comparable to mAP_0.5 from yolov5 training script. In icevision use print_summary = True to get all metrics

image