Giters

isarandi / metrabs

Estimate absolute 3D human poses from RGB images.

Home Page:https://arxiv.org/abs/2007.07227

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Finetuning the public models

fabianwindbacher opened this issue · comments

commented

Hi, very cool repo!

I am trying to fine-tune one of your public models (e.g. metrabs_mob3l_y4t) on a new dataset.
Since these are the packaged multi-person models, I extracted the crop model itself.

However, using the checkpoint system implemented in main.py, it seems like the weights are not copied.
log_detailed.txt:

WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).variables.0
WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).variables.1
...
WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).variables.265

Can you point me in the right direction?

commented

Hmm. Probably the variable names get messed up or lost after packaging into the multi-person SavedModel. For this use case it would be best to provide the original checkpoint files too. I'll look into this.