why the target_index variable is invalid in your code?

Caoliangjie / pytorch-gradcam-resnet50

CAM图的resnet50版本

150 stars 42 forks source link

why the target_index variable is invalid in your code? #10

Closed HulkMaker closed 5 years ago

HulkMaker commented 5 years ago

change the target_index number in main function, it's no difference for the cam mask. It mains we can only check the highest score category, or something wrong?

Caoliangjie commented 5 years ago

Hi muyiguangda, you can change to the deeper layer to see the diffences, it really works. Here's two pitures for example.(1 is set index to 100, 2 is None) Maybe this can help you.