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
131 stars 26 forks source link

Added feature to enable accessing outputs of the model. #24

Open schelbiup opened 3 weeks ago

schelbiup commented 3 weeks ago

Added feature to enable accessing outputs after the image is passed through EigenCAM. Let cam be an instance of EigenCAM and the image forwarded throught the model via cam(...), then the model outputs can be accessed via e.g., cam.outputs[0].probs.top5 or cam.outputs[0].names etc.