leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes
642 stars 259 forks source link

add evaluate.py #40

Open alvenchen opened 5 years ago

alvenchen commented 5 years ago

add evaluate.py according to val.ipynb. then make some changes on gpu/cpu switch, mat file reading and ground truth extracting.