Skip to content

Latest commit

 

History

History
61 lines (49 loc) · 2.34 KB

README.md

File metadata and controls

61 lines (49 loc) · 2.34 KB
Raw

Crowd-Counting-with-Deep-Negative-Correlation-Learning

The codes for CVPR-2018 paper "Crowd Counting with Deep Negative Correlation Learning" in http://openaccess.thecvf.com/content_cvpr_2018/papers/Shi_Crowd_Counting_With_CVPR_2018_paper.pdf

To run this codes, what you need to do is as follows:

  1. Compiling the Caffe codes. You should have installed Caffe correctly. Then you can clone or download our codes and make some changes in Makefile.config to compile correctly.

  2. Preparing your data. The codes for preparing crowd counting dataset can be found in ¨examples/crowd/shanghaiA/predata¨. If you would like to run your own tasks, you have to write the codes by yourself.

  3. Training. In ¨examples/crowd¨, you can find the network prototxt and solver prototxt, and you should make some changes according to your tasks. If you want to use different ¨K¨ (K stands for the number of base regresors in the ensemble, we found K=64 gives the best performance in our tasks.), the ¨examples/crowd/shanghaiA/create_prototxt.py¨ can help to generate your network prototxt easily.

  4. Testing. If you use MAE and MSE as your evaluation metrics, you can monitor the testing results in training. You just need to add ¨layer { name: "mae" type: "MAELoss" bottom: "avgscore" bottom: "label" top: "mae" include { phase: TEST } }¨ and ¨layer { name: "mse" type: "MSELoss" bottom: "avgscore" bottom: "label" top: "mse" include { phase: TEST } }¨ to your network prototxt.

  5. Please carefully tune learning rate, we found it had a great influence to the training results in our tasks.

If you use this codes, please kindly cite our paper:

 @inproceedings{Shi_2018_CVPR,
 title={Crowd Counting With Deep Negative Correlation Learning},
 author={Shi, Zenglin and Zhang, Le and Liu, Yun and Cao, Xiaofeng and Ye, Yangdong and Cheng, Ming-Ming and Zheng, Guoyan},
 booktitle={CVPR},
 year={2019}
 }     
 
 @articles{zhang2019tpamicounting,
 title={Nonlinear Regression via Deep Negative Correlation Learning},
 author={Le Zhang, Zenglin Shi, Ming-Ming Cheng, Yun Liu, Jia-Wang Bian, Joey Tianyi Zhou, Guoyan Zheng, Zeng Zeng},
 booktitle={TPAMI},
 year={2019}
 }

Please feel free to contact us if you still have any questions. Zenglin Shi: [email protected] Le Zhang: [email protected]