search

AliaksandrSiarohin / monkey-net

Animating Arbitrary Objects via Deep Motion Transfer
467 stars 81 forks source link

Extract keypoints #24

Open vatsalag99 opened 4 years ago

vatsalag99 commented 4 years ago

Hi, thanks for this work! I wanted to ask how one could extract the keypoints from the KeyPointDetector using the mean and variance outputs and map them on the source image.

AliaksandrSiarohin commented 4 years ago

Keypoints is the mean

vatsalag99 commented 4 years ago

Is it possible to visualize the keypoints on the original source image?

AliaksandrSiarohin commented 4 years ago

I visualize it during training using this function https://github.com/AliaksandrSiarohin/monkey-net/blob/0c8aa4ea458a79717d9016a36a3d86d10a166aea/logger.py#L97