fire717 / movenet.pytorch

A Pytorch implementation of MoveNet from Google. Include training code and pre-trained model.
MIT License
374 stars 87 forks source link

reg label 生成 #12

Open cassie101 opened 2 years ago

cassie101 commented 2 years ago

https://github.com/fire717/movenet.pytorch/blob/bbc81408bd4da49789d912fd08635355fe123e60/lib/data/data_tools.py#L123-L136

您好,感谢您的分享,看了这段关于reg生成heatmap的代码,是在cx,cy周围(左与上边2个pixel;右与下边3个pixel)也附上reg值,但在不同位置时调整偏移是根据 img_size//4/2-1,相当与图的一半。这样的操作导致比如 - cx,cy在图上半部分,reg值在cx,cy周围反而偏多或少了?

fire717 commented 2 years ago

你好,这个我记得当时好像是因为中心点在图像中心左侧、右侧以及上下侧取法都不一样,这会来看似乎又有点不对,我最近略忙,你可以先实验看看是否有提升。后面有空我也测试看看。