zhenhuat / STCFormer

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

您好,关于训练两个数据集的模型的问题 #14

Open dlnn123 opened 3 months ago

dlnn123 commented 3 months ago

您好,请问run_3dhp_stc.py和run_stc.py中引用两个模型?为啥不都使用stcformer.py呢?看起来差别还蛮大的

zhenhuat commented 3 months ago

您好,请问run_3dhp_stc.py和run_stc.py中引用两个模型?为啥不都使用stcformer.py呢?看起来差别还蛮大的

哈喽,run_3dhp_stc.py用的是原始版本的模型代码,多了很多冗余的模块。但是只是代码有区别,实际调用的模块没有区别,你可以跟着forward函数倒推。

dlnn123 commented 3 months ago

您好,请问run_3dhp_stc.py和run_stc.py中引用两个模型?为啥不都使用stcformer.py呢?看起来差别还蛮大的

哈喽,run_3dhp_stc.py用的是原始版本的模型代码,多了很多冗余的模块。但是只是代码有区别,实际调用的模块没有区别,你可以跟着forward函数倒推。

好的,也就是在run_3dhp_stc.py中直接from stcformer import Model也可以吗?

dlnn123 commented 3 months ago

您好,请问run_3dhp_stc.py和run_stc.py中引用两个模型?为啥不都使用stcformer.py呢?看起来差别还蛮大的

哈喽,run_3dhp_stc.py用的是原始版本的模型代码,多了很多冗余的模块。但是只是代码有区别,实际调用的模块没有区别,你可以跟着forward函数倒推。

好的,也就是在run_3dhp_stc.py中直接from stcformer import Model也可以吗?

直接不使用原始版本的模型代码