How can i compute Integrated Gradients for object detection(Yolov5) ?

[akashlp27]

Hey, how can I compute IG for object detection models such as yolov5 wherein the model's output will usually be (class, bboxes, score), where as certain classification models give prediction probability for all classes, any suggestions would help a lot...

[99warriors]

Hi @akashlp27 Thank you for the question - you could try wrapping your original model to produce predictions for a pre-specified bbox. This way the wrapped model produces the output that IG expects. You'll need to make a new wrapped model for each bbox.

[akashlp27]

Thanks for the reply, will try it out

[mshergad]

Hi @akashlp27 - were you able to do this? I have the exact same problem and I’m not sure how to pre specify bounding boxes in yolov5