search

jacobgil / pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
https://jacobgil.github.io/pytorch-gradcam-book
MIT License
10.3k stars 1.54k forks source link

YOLOv8n auto train with 100 epochs starts when in try to use the EigenCAM method #507

Open enricopierga opened 3 months ago

enricopierga commented 3 months ago

Hi, I want to use the EigenCAM technique on my YOLOv8n model, preaddestred for a detection task. Once i declare my YOLO .pt weights and I create my tensor image, I try to call the EigenCAM method to generate activation maps with my chosen target layers. Now here's the problem: a 100 epochs automatic training starts. Even when i put my model in .eval() mode this train starts... can someone please help me?

Here it is my image in tensors image

Here it's when my train starts image

These are my chosen target layers image

I followed step by step the tutorial with the pretrained YOLOv5 model (EigenCAM puppies) and all things works perfectly in there... @jacobgil

Andreavisi1 commented 3 months ago

same problem

atultiwari commented 2 months ago

I am also facing the same problem.