Skip to content

(ECCV'20) Weighing Counts: Sequential Crowd Counting by Reinforcement Learning

License

Notifications You must be signed in to change notification settings

poppinace/libranet

Repository files navigation

LibraNet

weighing_counts

This repository includes the official implementation of LibraNet for crowd counting, presented in our paper:

Weighing Counts: Sequential Crowd Counting by Reinforcement Learning

Proc. European Conference on Computer Vision (ECCV), 2020

Liang Liu1,†, Hao Lu2,†, Hongwei Zou1, Haipeng Xiong1, Zhiguo Cao1, Chunhua Shen1

1Huazhong University of Science and Technology

2The University of Adelaide, Australia

equal contribution

Model Structure

Installation

The code has been tested on Python 3.7.6 and PyTorch 1.4.0. Please follow the official instructions to configure your environment. See other required packages in requirements.txt.

Data Structure

$./data/
├──── Train
├──── Test

Training

$./backbone.pth.tar

Train LibraNet on SHT Part_A Dataset

python train.py

Inference

Pre-trained Model on SHT Part_A dataset

  • Download the model from: [BaiduYun (68.3 MB) (code: 20um) or OneDrive (70 MB)](https://1drv.ms/u/s!AkNf_IPSDakh8XBVTepnGq2J_YjN?e=lJCCUw)
  • The result of this model is: mae=55.5, mse=93.9. However, if the pythorch version is less than 1.4.0 (1.3.1 for example), the result might be: mae=56.3 , mse=95.2. Now I try to find the reason.
  • Move the model into the folder, and the path structure should like this:
$./trained_model/
├──── LibraNet_SHT_A.pth.tar

Evaluation

python Test_SHT_A.py

Citation

If you find this work or code useful for your research, please cite:

@article{liu2020WeighingCounts,
  title={Weighing Counts: Sequential Crowd Counting by Reinforcement Learning},
  author={Liu, Liang and Lu, Hao and Zou, Hongwei and Xiong, Haipeng and Cao, Zhiguo and Chun, Huashen},
  journal={Proc. Eur. Conf. Computer Vision},
  year={2020}
}

Update

2020-9-24

  1. Fix a bug in train_test.py line 32
  • Error:
for image_index in range(0, 1):
  • Correct:
for image_index in range(0, train_number):
  1. Add LICENSE.md

Permission

The code are only for non-commercial purposes. Copyrights reserved.

Contact: Liang Liu ([email protected]) Hao Lu ([email protected])

About

(ECCV'20) Weighing Counts: Sequential Crowd Counting by Reinforcement Learning

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages