Calculating a 'Detection Rate' | YOLOv5

[kyrangraves]

Hi All,

I have trained a bunch of YOLOv5 models with varying parameters to detect different coral morphologies from ROV imagery. I want to modify the val.py script to create an evaluation metric that determines the % of ground truth objects that have been correctly detected, irrespective of whether the assigned label (classification) is correct. If I'm correct, the out-the-box val.py script determines a true positive when your defined IoU threshold and correct classification are met. I'm essentially wanting to remove the classification element of this.

Before I try, has anybody written a bit of script like this before or know of another simpler way of calculating this metric?

All the best - Kyran