search

timothybrooks / instruct-pix2pix

Other
6.41k stars 542 forks source link

Visualizing Attention Maps #70

Open nmakes opened 1 year ago

nmakes commented 1 year ago

Hi,

Thanks for the wonderful implementation. I wanted to visualize the cross-attention maps between image and text query during inference. Could you please advice how I should go about it (e.g. the relevant module where I should put a callback)?

Thanks in advance.

victorzarzu commented 7 months ago

Hi @nmakes! Did you find a solution for this? :)