search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.42k stars 4.23k forks source link

Grad cam for yolov7 #519

Closed holger-prause closed 2 years ago

holger-prause commented 2 years ago

Hello, is there any grad cam available for yolov7 or any mechanism which helps me understanding for which features the ,model looks for? This would help a lot during trouble shooting / debugging.

As this is pytorch based, maybe there is already a solution out there , but i am bit worried about the additional nms layer and anchors(as i understand nms layer shouldnt interfer as you grab weights from earlier layers).

Thank you very much + greetings, Holger

WongKinYiu commented 2 years ago

For your reference. Not yet tested.

holger-prause commented 2 years ago

Thank you very much for the linked resources / proposed solutions.