人群计数:C3 Framework: An Open-source PyTorch Code for Crowd Counting

人群计数的第一个开源框架,基于pytorch实现。

框架中提供了六种主流的数据集预处理方案,分别是UCF CC 50 , World-Expo’10 , SHT A , SHT B , UCF-QNRF ,and GCC。

框架对于输入图片的尺寸,会限制其高度和宽度,使其能被16整除,以确保一些下采样层可以输出想要的尺寸。
在这里插入图片描述

对于bitch size的大小,有预训练模型的会固定一个值,没有的就用多个值。顾及到数据集中的不同尺寸的图片,框架采用N(bitch size) * 3 * min(h) * min(w)作为输入张量的大小。

对于标签的两种操作:

  1. ground truth scale down-sampling
  2. label normalization(to 100).

提供了三种预训练模型:AlexNet,VGG,ResNet. 并针对每一个都做了一些修改。

文中复现了4个模型:MCNN, CMTL,CSRNet,SANet.
在这里插入图片描述

这是作者大佬的自述(“自述”这个词貌似用得不太对):https://zhuanlan.zhihu.com/p/65650998

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值