SASNet(AAAI2021)
宋庆宇*,王长安*,王亚标,应泰,王成杰,李吉林,吴健,马嘉宜在SASNet的PyTorch中正式实现,如“选择还是融合?人群选择的规模选择”中所述。
该代码已通过PyTorch 1.5.0进行了测试。 它可能无法与其他版本一起运行。
比例尺选择的可视化
提出的自适应选择策略可自动学习内部关系,以下可视化效果证明了其有效性。
安装
将此仓库克隆到名为SASNet_ROOT的目录中
安装Python依赖项。 我们使用python 3.6.8和pytorch> = 1.5.0
pip install -r requirements.txt
从下载ShanghaiTech数据集和模型
准备
按以下方式组织数据和模型:
SASNet_ROOT/
|->datas/
| |->part_A_final/
2022-07-15 18:55:10
29.01MB
Python
1