Visualizer是一个辅助深度学习模型中Attention模块可视化的小工具,主要功能是帮助取出嵌套在模型深处的Attention Map(也可以提取其他变量)
使用方法:
1、按照GitHub教程安装(Visualizer)
2、之后在/data2/lqh/workspace_pycharm/OSTrack/lib/test/tracker/ostrack_sam_dConv.py
的第一行加入
from visualizer import get_local
get_local.activate() # 激活装饰器
3、### Usage1
[](https://github.com/luo3300612/Visualizer#usage1)
比如说,我想要函数里的attention_map
变量: 在模型文件里,我们这么写
from visualizer import get_local
@get_local('attention_map')
def your_attention_function(*args, **kwargs):
...
attention_map = ...
...
return ...
1 条评论
全球视野与本土实践结合恰到好处。