Efficient-AI-Backbones
Efficient-AI-Backbones copied to clipboard
Visualization in VIG
Hi ,
Very impressive paper. Could you guide me to the code of visualization in your paper. I was trying to find it in the repo and was not able to find it.
- Run inference and get the indice of k-NN: https://github.com/huawei-noah/Efficient-AI-Backbones/blob/f4ffbe5fa41934ae98768bc7c720faf5a661b0b4/vig_pytorch/gcn_lib/torch_vertex.py#L128
- Draw the centroid and its k-NN nodes using any tool like matplolib or simply PPT.
hi, why the visualization.ipydb file disappear?