search

AliaksandrSiarohin / pose-gan

381 stars 78 forks source link

How to deal with missing joint? #60

Closed chmxu closed 4 years ago

chmxu commented 4 years ago

Thanks for the greate work! I want to know what if the human body has some missing joints? It seems the code restricts some joints to be valid. For example, when i don't have hip/shoulder joints, the function in https://github.com/AliaksandrSiarohin/pose-gan/blob/8693ec4b69fd3e83948bb1285c8e828fc62da390/pose_transform.py#L101 could raise KeyError.

AliaksandrSiarohin commented 4 years ago

Yes, the model assume that body is detected. You will not get any decent output from images like this, if you need some output inpaint missing joint with something, for example average location of this joint.

chmxu commented 4 years ago

I see. Thanks again.