search

chenyilun95 / tf-cpn

Cascaded Pyramid Network for Multi-Person Pose Estimation (CVPR 2018)
MIT License
793 stars 197 forks source link

About heatmap generation #32

Closed weleen closed 6 years ago

weleen commented 6 years ago

https://github.com/chenyilun95/tf-cpn/blob/7c01aba69b8b9ad14fbb36d409f850bafd345dd5/models/COCO.res50.256x192.CPN/dataset.py#L156 I am curious about the generation of heatmap, why multiply 255 here? It seems that the network can not converge when i remove 255.

chenyilun95 commented 6 years ago

The prediction heatmap is expected to range in [0...255], which affects the network loss. I think it works if you increase learning rate properly and remove 255.