maohule/PCEDNet

PCEDNet

Introduction

This is the official PyTorch implementation of paper: Progressive Crowd Enhancement De-Background Network for Crowd Counting.

Prepare data

Our processed dataset can be download here: google drive

Pre-trained model

Download pre-trained model files: 百度云网盘，8c1t

Train

python train.py

Test

python test.py