Awesome
Introduction
This is the code repository for the Heatmap Regulation (HR) paper Improving Object Counting with Heatmap Regulation. Download the pretrained models from this link.
Note that the GAP-GAS models actually refer to the HR models in this repo.
Prerequisites
- Python2.7
- PyTorch v0.2.0_3 (Models are trained on this version)
Code Overview
Codes for each of the datasets are into separate sub-directories inside the 'codes/' directory. Also, put all datasets in the corresponding sub-directories inside 'data/' directory. Please follow the readme inside each code sub-directory to train and test the models.