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

Request for Example Complete Notebook - YOLO v8 CAM #9

Closed snehitvaddi closed 6 months ago

snehitvaddi commented 7 months ago

Hello,

I'm conducting research on Explainable AI using YOLO v8 with a custom dataset and encountered some uncertainties in the documentation. My queries are:

  1. What should be specified in the "model" parameter?
  2. What is the appropriate input for the "target_layers" attribute?

A detailed example or a notebook demonstrating end-to-end implementation on a sample dataset would be greatly beneficial, especially for beginners like me😬. This could really jazz up the repo!

Best, Snehit Vaddi

snehitvaddi commented 6 months ago

@rigvedrs I am still waiting for a response on this. Would greatly appreciate if you could reply something.