You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
chihy
d53808c9d0
|
2 years ago | |
|---|---|---|
| .idea | 2 years ago | |
| __pycache__ | 2 years ago | |
| data | 2 years ago | |
| datasets | 2 years ago | |
| doc | 2 years ago | |
| losses | 2 years ago | |
| models | 2 years ago | |
| utils | 2 years ago | |
| vgg | 2 years ago | |
| README.md | 2 years ago | |
| __init__.py | 2 years ago | |
| demo.py | 2 years ago | |
| preprocess_dataset.py | 2 years ago | |
| test_1.py | 2 years ago | |
| train.py | 2 years ago | |
| train.txt | 2 years ago | |
| val.txt | 2 years ago | |
| vgg19-fpn-new.pth | 2 years ago | |
README.md
项目中文名
多尺度贝叶斯人群计数
项目英文名
Multi-Scale-Bayesian-Crowd-Counting
作者
Zhiheng Ma, Xing Wei, Xiaopeng Hong, Yihong Gong
项目简介
1. 功能
人群计数
2. 性能
在UCF-QNRF上达到当前最优性能
在UCF-QNRF上 MAE = 86.50；MSE = 146.55;
3. 评估指标
MAE MSE
4. 使用数据集
UCF-QNRF
https://www.crcv.ucf.edu/data/ucf-qnrf/
运行环境与依赖
|类别|名称|版本|
|os|ubuntu|14.04|
|深度学习框架|pytorch|1.0.0|
||opencv|3.4.9|
输入与输出
|名称|说明|
|输入|RGB图像|
|输出|多尺度热力图|计数值|
运行方式
在terminal下运行以下命令。
python preprocess_dataset.py --origin_dir <directory of original data> --data_dir <directory of processed data>
python train.py --data_dir <directory of processed data> --save_dir <directory of log and model>
python test.py --data_dir <directory of processed data> --save_dir <directory of log and model>