Hi, thank you for your great job!

I tried to use a custom model, and even though the object is detected, the heatmap is basically empty.

Do you know a workaround?

Thanks agin.

Dears

When I try to operate "python main.py --model-path yolov5l.pt --img-path images/cat-dog.jpg --output-dir outputs" and I get this error

Traceback (most recent call last): File "main.py", line 6, in from models.yolo_v5_object_detector import YOLOV5TorchObjectDetector File "/content/yolov5-gradcam/models/yolo_v5_object_detector.py", line 2, in from deep_utils.utils.box_utils.boxes import Box File "/usr/local/lib/python3.7/dist-packages/deep_utils/init.py", line 10, in from .vision import * File "/usr/local/lib/python3.7/dist-packages/deep_utils/vision/init.py", line 1, in from .face_detection import * File "/usr/local/lib/python3.7/dist-packages/deep_utils/vision/face_detection/init.py", line 49, in from deep_utils.vision.face_detection.ultralight.tf.ultralight_tf_face_detection import ( File "/usr/local/lib/python3.7/dist-packages/deep_utils/vision/face_detection/ultralight/tf/ultralight_tf_face_detection.py", line 17, in from .utils.rfb_320 import create_rfb_net File "/usr/local/lib/python3.7/dist-packages/deep_utils/vision/face_detection/ultralight/tf/utils/rfb_320.py", line 1, in import tensorflow as tf File "/usr/local/lib/python3.7/dist-packages/tensorflow/init.py", line 37, in from tensorflow.python.tools import module_util as _module_util File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/init.py", line 37, in from tensorflow.python.eager import context File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/eager/context.py", line 29, in from tensorflow.core.framework import function_pb2 File "/usr/local/lib/python3.7/dist-packages/tensorflow/core/framework/function_pb2.py", line 16, in from tensorflow.core.framework import attr_value_pb2 as tensorflow_dot_core_dot_framework_dot_attr__value__pb2 File "/usr/local/lib/python3.7/dist-packages/tensorflow/core/framework/attr_value_pb2.py", line 16, in from tensorflow.core.framework import tensor_pb2 as tensorflow_dot_core_dot_framework_dot_tensor__pb2 File "/usr/local/lib/python3.7/dist-packages/tensorflow/core/framework/tensor_pb2.py", line 16, in from tensorflow.core.framework import resource_handle_pb2 as tensorflow_dot_core_dot_framework_dot_resource__handle__pb2 File "/usr/local/lib/python3.7/dist-packages/tensorflow/core/framework/resource_handle_pb2.py", line 150, in 'module' : 'tensorflow.core.framework.resource_handle_pb2' SystemError: google/protobuf/pyext/descriptor.cc:358: bad argument to internal function

Could you tell me how to solve this error?

I took the gradcam for yolov3 and revise a little. I have no idea about why register_backward_hook() couldn't get the gradient. The register_forward_hook() is ok. can you help me, thanks a lot.

D:\anaconda\envs\yolov5\lib\site-packages\deep_utils\vision\face_detection\ultralight_init_.py:3: UserWarning: No module named 'deep_utils.vision.face_detection.ultralight.torch'. If you don't use UltralightTorchFaceDetector ignore this message. UltralightTorchFaceDetector = import_module( D:\anaconda\envs\yolov5\lib\site-packages\deep_utils\vision\face_detection\ultralight_init_.py:6: UserWarning: No module named 'deep_utils.vision.face_detection.ultralight.tf'. If you don't use UltralightTFFaceDetector ignore this message. UltralightTFFaceDetector = import_module( D:\anaconda\envs\yolov5\lib\site-packages\deep_utils\callbacks\tf_keras_init_.py:3: UserWarning: No module named 'tensorflow'. If you don't use LRScalar ignore this message. LRScalar = import_module( [INFO] Loading the model Traceback (most recent call last): File "main.py", line 125, in main(args.img_path) File "main.py", line 64, in main model = YOLOV5TorchObjectDetector(args.model_path, device, img_size=input_size, File "D:\AAAAAAAAA\YOLOV5-4.0\yolov5-gradcam-master\models\yolo_v5_object_detector.py", line 32, in init self.model = attempt_load(model_weight, device=device) File "D:\AAAAAAAAA\YOLOV5-4.0\yolov5-gradcam-master\models\experimental.py", line 80, in attempt_load ckpt = torch.load(attempt_download(w), map_location=device) File "D:\anaconda\envs\yolov5\lib\site-packages\torch\serialization.py", line 594, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "D:\anaconda\envs\yolov5\lib\site-packages\torch\serialization.py", line 853, in _load result = unpickler.load() File "D:\anaconda\envs\yolov5\lib\site-packages\torch\serialization.py", line 845, in persistent_load load_tensor(data_type, size, key, _maybe_decode_ascii(location)) File "D:\anaconda\envs\yolov5\lib\site-packages\torch\serialization.py", line 834, in load_tensor loaded_storages[key] = restore_location(storage, location) File "D:\anaconda\envs\yolov5\lib\site-packages\torch\serialization.py", line 811, in restore_location return default_restore_location(storage, map_location) File "D:\anaconda\envs\yolov5\lib\site-packages\torch\serialization.py", line 178, in default_restore_location raise RuntimeError("don't know how to restore data location of " RuntimeError: don't know how to restore data location of torch.HalfStorage (tagged with )

I added a feature so that --img-path can accept folder as path. So, if it is folder, it will loop through all files and interpret them one by one.

This is default functionality in YOLO.

Hi, Thanks for your great work and the results look amazing. But I wonder that if it can visualize multi-classes by grad-cam whatever in classification or detection task.

hello, I trianed it on DOTA datasets,but get error. RuntimeError: The size of tensor a (80) must match the size of tensor b (1280) at non-singleton dimension 1

Hi there,

I'm currently have issues when running the code with the model I trained on yolov5 v6.1, with a custom dataset Do you know what's happening?

Thank you so much!!

Here's my code...↓

! pip install -r requirements.txt

import os
os.chdir("./yolov5-gradcam-master/")

!python main.py --model-path runs/train/hat_person4/weights/best.pt --img-path datasets/hat_person/yolo_dataset/images/train/000490.jpg --output-dir outputs --names hat,person   

And then...Something wrong..

[INFO] Loading the model Traceback (most recent call last): File "main.py", line 124, in main(args.img_path) File "main.py", line 64, in main names=None if args.names is None else args.names.strip().split(",")) File "f:\works\VSCODE\yolov5\yolov5-gradcam-master\models\yolo_v5_object_detector.py", line 32, in __init__ self.model = attempt_load(model_weight, device=device) File "f:\works\VSCODE\yolov5\yolov5-gradcam-master\models\experimental.py", line 80, in attempt_load ckpt = torch.load(attempt_download(w), map_location=device) File "d:\Anaconda3\envs\yolo\lib\site-packages\torch\serialization.py", line 607, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "d:\Anaconda3\envs\yolo\lib\site-packages\torch\serialization.py", line 882, in _load result = unpickler.load() File "d:\Anaconda3\envs\yolo\lib\site-packages\torch\serialization.py", line 875, in find_class return super().find_class(mod_name, name) AttributeError: Can't get attribute 'DetectionModel' on <module 'models.yolo' from 'f:\works\VSCODE\yolov5\yolov5-gradcam-master\models\yolo.py'>

hi,i tried to use the yolov5l,yolov5x,yolov5s models,but the results of heatmap were not good,although i tried other targets-layer. please tell me how this work support yolov5l,yolov5x,yolov5s？thanks