C3Framework是一个开源的人群计数框架,在进行代码复现的时候,进行单张图片的测试。
1. C3Framework中的算法
根据官网介绍,在该框架中集成了若干个模型,其中有些模型的复现效果并没有论文上来得好。
Method | WE | UCF50 |
---|---|---|
MCNN (RGB Image) | TBD | TBD |
AlexNet (conv5) | TBD | TBD |
VGG-16 (conv4_3) | TBD | TBD |
VGG-16 (conv4_3)+decoder | TBD | TBD |
ResNet-50 (layer3) | TBD | TBD |
ResNet-101 (layer3) | TBD | TBD |
CSRNet | TBD | TBD |
SANet | TBD | TBD |
CMTL | TBD | TBD |
ResSFCN-101 (SFCN+) | TBD | TBD |
2. C3Framework中的算法测试
撰写单个图片的测试代码如下,在该代码中同时进行了MAE和MSE的计算。
以数据集中的第一张图片来测试:
其原始的人群密度图如下:
推理得到的结果如下:
其计算出来的MAE和MSE大致在21左右。