search

HowieMa / TransFusion-Pose

[BMVC2021] "TransFusion: Cross-view Fusion with Transformer for 3D Human Pose Estimation"
MIT License
40 stars 5 forks source link

pkl in pre training model #7

Closed jianlai123-123 closed 1 year ago

jianlai123-123 commented 1 year ago

First of all, thank you for your contribution and sharing. I'm very interested in your project. I'd like to ask about the PKL files in the pre training. What is the function of the PKL file? Can I use it directly? Look forward to your reply, thank you again! @HowieMa

HowieMa commented 1 year ago

May I ask which pkl file you refer to? If you are talking about the pkl file in the annotation, you can obtain it from the preprocessing of Human 3.6M.

image

jianlai123-123 commented 1 year ago

Hello, thank you for your reply. What I want to consult is the pkl file obtained from the download path. That is the document generated by the operation of other projects, and has nothing to do with this project, right? What I want to ask is the one marked in green.

333 334

@HowieMa

HowieMa commented 1 year ago

Hi, I am sorry that I don't know how you get this data.pkl file. If you directly download the files from the provided Google Drive link, you should not have it. Here are all files inside the output folder. image

The file "model_best_pth.tar" is the weights for the trained model, which is obtained withtorch.save(...) (see this code for more detail) You don't need to unzip it.

jianlai123-123 commented 1 year ago

Thank you for your reply.