search

leeyeehoo / CSRNet-pytorch

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

What needs to be changed if the model is not loaded with VGG pretrained? #80

Open Brother-Q opened 3 years ago

Brother-Q commented 3 years ago

I tried not to load the pre-training weights of VGG16 and got a MAE more than 200. How do I adjust the parameters in training? Has anyone tried to replace other backbone networks to extract crowd density information?

raisinglc commented 3 years ago

I think there is not much thing which can help you to improve the performance. The training set is very small. It's nearly impossible to achieve the same result without loading pretrained model. If you use a more complicate model, it might perform will, but i think it cannot achieve the same level. Moreover, I argue CSRNet. The model's head has big shortcoming.