I have added other modules in the yolov8 framework, can I generate heat maps with your repository code?

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

I have added other modules in the yolov8 framework, can I generate heat maps with your repository code? #15

Closed weldingCode closed 1 month ago

rigvedrs commented 1 month ago

You can give it a try. It basically should give an output considering it will just resize the output matrix of activations from the layer into the image size, and generate the heatmap from it.