target_layer path error about torch-cam HOT 3 CLOSED

Thank you!
code: cam_extractor = CAM(model, target_layer=model.backbone.shared_module_bh.model_sh_fr.layer4[-1].conv3, fc_layer=model.classifier) can work. The network structure should not add ' ' .

from torch-cam.

Hello @ReIDkj 👋

Yes there is! Here is the explanation:

• you can specify the target layer using a string argument or the module itself
• when using the string version, the underneath mechanism uses the key of the state_dict which replaces index brackets [0] by a dot .0 (bare in mind that -1 won't work then

So you can either use the module itself (recommended):

cam_extractor = CAM(model, model.backbone.shared_module_bh.model_sh_fr.layer4[-1]) 

or use the string argument (below I supposed that the last item was at index 3):

cam_extractor = CAM(model, 'model.backbone.shared_module_bh.model_sh_fr.layer4.3') 

Hope this helps, let me know if some aspects are unclear :)
Cheers ✌️

from torch-cam.

Happy to help! I'll close the issue then, feel free to reopen if some details are unclear :)

from torch-cam.