Code on the visualization of attention maps

johnrso / spawnnet

SpawnNet: Learning Generalizable Visuomotor Skills from Pre-trained Networks

https://xingyu-lin.github.io/spawnnet

MIT License

34 stars 3 forks source link

Code on the visualization of attention maps #4

Open yolo01826 opened 2 months ago

yolo01826 commented 2 months ago

Hello, author. Thank you very much for your work. I noticed that you have shared the heat map that has been processed. If I want to use spawn-net for my own task, how can I visualize the attention map to adjust the strategy? Can you share the relevant code? @johnrso

yolo01826 commented 2 months ago

@Xingyu-Lin