Home

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

  1. Python2.7
  2. 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.