Closed infochen closed 2 years ago
I used yolov3.pt for the yolov5-gradcam. The result like this:
I didn't change anything this time. I am confused about this situation
I figured out! I choose wrong layer before and the choosed layer' size must small.
Hello, I have the same problem, may I ask how to solve it
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.![image](https://user-images.githubusercontent.com/100501692/158021281-b6f32de2-e94d-4ae7-b564-b0f94eb69102.png)