search

hustvl / YOLOS

[NeurIPS 2021] You Only Look at One Sequence
https://arxiv.org/abs/2106.00666
MIT License
827 stars 118 forks source link

Visualization demo? #11

Open lucasjinreal opened 2 years ago

lucasjinreal commented 2 years ago

How to inference on a single image and visualize the result?

Yuxin-CV commented 2 years ago

Try the notebook.

nomurakeiya commented 2 years ago

@Yuxin-CV Can you tell me more?

Yuxin-CV commented 2 years ago

@Yuxin-CV Can you tell me more?

Try https://github.com/hustvl/YOLOS/blob/main/VisualizeAttention.ipynb