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 ...
最后修改:2024 年 08 月 28 日
如果觉得我的文章对你有用,请随意赞赏