Thanks for the reply!
I no longer get the KeyError but I know was wondering if you could help me visualize the feature map. Like you said the imshow doesn’t work.
The code now looks similar to something you wrote on a different toppic: Visualize feature map
So what would you think would be the best way to visualize the feature map for a given image?