search

rigvedrs / YOLO-V8-CAM

Wanna know what your model sees? Here's a package for applying EigenCAM on the new YOLO V8 model
MIT License
100 stars 17 forks source link

gradcam for yolov8-seg #5

Closed yebuxiu123 closed 4 months ago

yebuxiu123 commented 10 months ago

Thank you very much for your work, but I am now in great need of gradcam support for yolov8 instance segmention, when will this be possible please!

enickel00 commented 7 months ago

I'm also interested if anyone's successfully implemented this for segmentation models with yolov8, or has any advice on how to get it going.

rigvedrs commented 6 months ago

Hey! I actually had no plans for this since I wanted to work only on Object detection and classification for my case. Thought that someone may figure it out soon. But since that's not the case I guess, I will definitely give it a try and let you know soon :)

Eric-Marchetti commented 5 months ago

Hello, thank you for working on this project! I noticed that yesterday you made a commit that added support for Segmentation models and added an example to the tutorial ipynb. However when looking at the base_cam.py file that was in that commit I don't see the segmentation support. Could you please either point me to where the code is or push it to the repo? Thank you!

rigvedrs commented 4 months ago

Hello, thank you for working on this project! I noticed that yesterday you made a commit that added support for Segmentation models and added an example to the tutorial ipynb. However when looking at the base_cam.py file that was in that commit I don't see the segmentation support. Could you please either point me to where the code is or push it to the repo? Thank you!

It has been provided in this notebook

enickel00 commented 4 months ago

@rigvedrs Thanks a bunch for doing that. Looking forward to trying it out.