文件名称:SA-Net:ICASSP 2021论文的代码
文件大小:72KB
文件格式:ZIP
更新时间:2024-05-24 19:23:54
Python
SA-Net:深度卷积神经网络的洗牌注意() 张庆龙,杨玉斌 [南京大学新型软件技术国家重点实验室] 方法 图1:洗牌注意模块图。 影像分类 我们提供在ImageNet-1k上预训练的基线sa_resnet模型。 名称 acc @ 1 #params(M) 网址 sa_resnet50 77.88 25.56 sa_resnet101 78.95 44.55 评估 要使用单个GPU运行在ImageNet val上评估经过预训练的sa_resnet50,请执行以下操作: python main.py -a sa_resnet50 -e --resume /path/to/sa_resnet50.pth.tar /path/to/imagenet 这应该给 * Acc@1 77.882 Acc@5 93.892
【文件预览】:
SA-Net-main
----figures()
--------sa.png(65KB)
----utils()
--------misc.py(1KB)
--------__init__.py(20B)
--------__pycache__()
----models()
--------__init__.py(24B)
--------sa_resnet.py(8KB)
--------__pycache__()
----cam()
--------__init__.py(31B)
--------group_act.py(2KB)
----main.py(16KB)
----README.md(1KB)