zhenhuat / STCFormer

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

可以分享一下3DHP的训练参数吗?感谢! #4

Open JihuaPeng opened 1 year ago

JihuaPeng commented 1 year ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

zhenhuat commented 1 year ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

哈喽,直接运行以下脚本就可以了。如果不行的话麻烦您随时跟我反馈,我有时间再看看。 python run_3dhp_stc.py --train 0 --frames 81 -b 128 -s 1 --reload 1 --previous_dir ./checkpoint/model_81_STMO/no_refine_8_2310.pth

JihuaPeng commented 1 year ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

哈喽,直接运行以下脚本就可以了。如果不行的话麻烦您随时跟我反馈,我有时间再看看。 python run_3dhp_stc.py --train 0 --frames 81 -b 128 -s 1 --reload 1 --previous_dir ./checkpoint/model_81_STMO/no_refine_8_2310.pth

是训练时候的参数设置呢,不是evaluate的,训练时候的参数设置也跟上面一样吗

zhenhuat commented 1 year ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

哈喽,直接运行以下脚本就可以了。如果不行的话麻烦您随时跟我反馈,我有时间再看看。 python run_3dhp_stc.py --train 0 --frames 81 -b 128 -s 1 --reload 1 --previous_dir ./checkpoint/model_81_STMO/no_refine_8_2310.pth

是训练时候的参数设置呢,不是evaluate的,训练时候的参数设置也跟上面一样吗

训练脚本把train 改成1,再把后面的reload 模型删掉就好了,如下: python run_3dhp_stc.py --train 1 --frames 81 -b 128 -s 1

JihuaPeng commented 1 year ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

哈喽,直接运行以下脚本就可以了。如果不行的话麻烦您随时跟我反馈,我有时间再看看。 python run_3dhp_stc.py --train 0 --frames 81 -b 128 -s 1 --reload 1 --previous_dir ./checkpoint/model_81_STMO/no_refine_8_2310.pth

是训练时候的参数设置呢,不是evaluate的,训练时候的参数设置也跟上面一样吗

训练脚本把train 改成1,再把后面的reload 模型删掉就好了,如下: python run_3dhp_stc.py --train 1 --frames 81 -b 128 -s 1

感谢感谢!

JihuaPeng commented 1 year ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

哈喽,直接运行以下脚本就可以了。如果不行的话麻烦您随时跟我反馈,我有时间再看看。 python run_3dhp_stc.py --train 0 --frames 81 -b 128 -s 1 --reload 1 --previous_dir ./checkpoint/model_81_STMO/no_refine_8_2310.pth

是训练时候的参数设置呢,不是evaluate的,训练时候的参数设置也跟上面一样吗

训练脚本把train 改成1,再把后面的reload 模型删掉就好了,如下: python run_3dhp_stc.py --train 1 --frames 81 -b 128 -s 1

这个参数设置占用的显存很大,请问是在A100上训练的吗

zhenhuat commented 1 year ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

哈喽,直接运行以下脚本就可以了。如果不行的话麻烦您随时跟我反馈,我有时间再看看。 python run_3dhp_stc.py --train 0 --frames 81 -b 128 -s 1 --reload 1 --previous_dir ./checkpoint/model_81_STMO/no_refine_8_2310.pth

是训练时候的参数设置呢,不是evaluate的,训练时候的参数设置也跟上面一样吗

训练脚本把train 改成1,再把后面的reload 模型删掉就好了,如下: python run_3dhp_stc.py --train 1 --frames 81 -b 128 -s 1

这个参数设置占用的显存很大,请问是在A100上训练的吗

哈喽,81帧不需要非常大的内存,但是具体的卡我不记得是哪种型号了(P40或V100),我只有243帧在A100上训练。您可以尝试降低batch_size大小看看效果。

LYee0515 commented 6 months ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

你好,我想请问您是如何处理3dhp数据集的,我使用P-STMO的方法处理之后 npz格式数据集大小只有331字节,方便的话可以发送给我一份数据集吗

zhenhuat commented 6 months ago

很棒的一个工作,可以分享一下3DHP的训练参数设置吗?感谢!

你好,我想请问您是如何处理3dhp数据集的,我使用P-STMO的方法处理之后 npz格式数据集大小只有331字节,方便的话可以发送给我一份数据集吗

哈喽,这是3DHP的数据链接:https://pan.baidu.com/s/1WZcOH3bP4UCoYeokVpnZyA 提取码:3dhp

jokerxxw commented 3 months ago

您好,我想请问一下,你的3dhp的训练参数,里面的采样数是不是3呀,我看human3.6都是基本上都是3