Efficient-AI-Backbones icon indicating copy to clipboard operation
Efficient-AI-Backbones copied to clipboard

Visualization in VIG

Open shubhrsingh22 opened this issue 3 years ago • 4 comments

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.

shubhrsingh22 avatar Sep 11 '22 10:09 shubhrsingh22

  1. 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
  2. Draw the centroid and its k-NN nodes using any tool like matplolib or simply PPT.

iamhankai avatar Sep 11 '22 12:09 iamhankai

hi, why the visualization.ipydb file disappear?

bang123-box avatar Jun 28 '23 03:06 bang123-box