上传者: 42178688
|
上传时间: 2022-03-24 09:18:45
|
文件大小: 10KB
|
文件类型: -
Pyramid_Scale_Network
这是PyTorch版本的“ repo”版本库,“利用多列架构在人群计数中的潜力”,它提供了用于人群计数任务的最先进,直接且端到端的体系结构。 我们还建议另一项有关人群计数的工作(),已被ECAI2020接受。
数据集
上海科技数据集
先决条件
我们强烈建议Anaconda作为环境。
的Python:3.6
PyTorch的:1.5.0
培训与测试
1,python make_dataset.py#生成基本事实。 ShanghaiTech数据集应放在“数据集”目录中。 2,python train.py#火车模型3,python val.py#测试模型
结果
partA:MAE 55.5 MSE 90.1
B部分:MAE 6.8 MSE 10.7