search

zhenhuat / STCFormer

(CVPR2023)3D Human Pose Estimation with Spatio-Temporal Criss-cross Attention
85 stars 5 forks source link

你好作者，关于MPJPE与p1 #13

Open dlnn123 opened 3 months ago

dlnn123 commented 3 months ago

你好作者，很棒的工作。请问一下在训练3DHP时候，logging.info('epoch: %d, lr: %.7f, loss: %.4f, MPJPE: %.2f, p1: %.2f' % (epoch, lr, loss, mpjpe, p1))中的p1是mpjpe还是pa-mpjpe？另外p1对应的是step里返回的 error_sum_test.avg？

zhenhuat commented 3 months ago

你好作者，很棒的工作。请问一下在训练3DHP时候，logging.info('epoch: %d, lr: %.7f, loss: %.4f, MPJPE: %.2f, p1: %.2f' % (epoch, lr, loss, mpjpe, p1))中的p1是mpjpe还是pa-mpjpe？另外p1对应的是step里返回的 error_sum_test.avg？

这里的MPJPE是训练集的结果： image,对应step函数中的第二个返回值 image

p1是指测试集的mpjpe： image，对应step函数中的唯一返回值 image

dlnn123 commented 3 months ago

明白了，谢谢您~ 我还有一个疑惑：opt.MAE,opt.refine在程序中默认都是false，是不是不需要管它俩？但是它俩都各自起到什么作用呢，要是为true的话

zhenhuat commented 3 months ago

这个代码是建立在P-STMO https://github.com/paTRICK-swk/P-STMO 上的，它采用了MAE来做预训练，也就是遮挡一部分2D关节点再尝试预测这些2D关节点；但是STCFormer并没有这样做，所以全是False。 opt.refine 可以参考这个 https://github.com/Vegetebird/StridedTransformer-Pose3D 代码，是一个转换坐标系后处理的模块。

dlnn123 commented 3 months ago

这个代码是建立在P-STMO https://github.com/paTRICK-swk/P-STMO 上的，它采用了MAE来做预训练，也就是遮挡一部分2D关节点再尝试预测这些2D关节点；但是STCFormer并没有这样做，所以全是False。 opt.refine 可以参考这个 https://github.com/Vegetebird/StridedTransformer-Pose3D 代码，是一个转换坐标系后处理的模块。

好的，谢谢您