search

heng-hw / SpaCap3D

[IJCAI 2022] Spatiality-guided Transformer for 3D Dense Captioning on Point Clouds (official pytorch implementation)
https://spacap3d.github.io/
20 stars 5 forks source link

Ask for visualizing colorful vector #8

Open jkstyle2 opened 4 months ago

jkstyle2 commented 4 months ago

Dear authors, Thanks for sharing your great work :) I was impressed how you depicts the eight-head attention values with colorful vectors as below. Could you share how we can visualize these vectors, if it is not manually done? image

Thanks for your help in advance!

heng-hw commented 1 month ago

Hi @jkstyle2! Thanks for your interest in our work. We follow this approach https://github.com/jessevig/bertviz to visualize the attention. Hope it helps!