Kernel Restarting The kernel for pytorch_yolov8/code/YOLO-V8-CAM-main/YOLO v8n EigenCAM.ipynb appears to have died. It will restart automatically.

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

Kernel Restarting The kernel for pytorch_yolov8/code/YOLO-V8-CAM-main/YOLO v8n EigenCAM.ipynb appears to have died. It will restart automatically. #17

Open Chen-chang-yu opened 2 months ago

Chen-chang-yu commented 2 months ago

when I run this line: grayscale_cam = cam(rgb_img) it appears this error "Kernel Restarting The kernel for pytorch_yolov8/code/YOLO-V8-CAM-main/YOLO v8n EigenCAM.ipynb appears to have died. It will restart automatically. " Could someone help me to solve this problem?? thx!

rigvedrs commented 1 month ago

This happens if your model is overloading your system. Are you able to run the model on it directly?

Chen-chang-yu commented 1 month ago

I tried this code with another computer equipped with RTX-4070TI, it works!! But my previous computer equiped with RTX-3060TI still failed.