Skip to content

Deep Density-aware Count Regressor: a state-of-the-art method for crowd counting

Notifications You must be signed in to change notification settings

GeorgeChenZJ/deepcount

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

deepcount

Deep Density-aware Count Regressor

PaddlePaddle Implementation Branch

https://github.com/GeorgeChenZJ/deepcount/tree/paddle

Installation

  1. Install requirements: Python 2.7, Tensorflow >= 1.8, PyTorch >= 0.4, Pillow, tqdm, scipy
  2. Download pretrained vgg model from https://download.pytorch.org/models/vgg16-397923af.pth
  3. Place vgg16-397923af.pth under the working directory

Data

  • Prepare the dataset first and modify data.py to make the data accessible. See data.py for detail. The current code of data.py is for Shanghai Tech Part B. Before training the model download Shanghai Tech dataset should be downloaded (https://drive.google.com/file/d/16dhJn7k4FWVwByRsQAEpl9lwjuV03jVI/view). Change 'data_dir' in data.py and relevant codes for loading the dataset.

Run

  • Set batch size and number of devices in train.py and then run
python train.py
  • Note that the code is written to be plain but not optimally efficient.

Trained Weights

About

Deep Density-aware Count Regressor: a state-of-the-art method for crowd counting

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages