search

pooya-mohammadi / yolov5-gradcam

Visualizing Yolov5's layers using GradCam
MIT License
281 stars 47 forks source link

Error when trying with custom model and dataset #24

Open bzha5848 opened 2 years ago

bzha5848 commented 2 years ago

Hi there,

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

Thank you so much!!