How can I use the checkpoint of first_order_motion_model with changing the number of segments

AliaksandrSiarohin / motion-cosegmentation

Reference code for "Motion-supervised Co-Part Segmentation" paper

Other

653 stars 142 forks source link

How can I use the checkpoint of first_order_motion_model with changing the number of segments #22

Open xupp1989 opened 4 years ago

xupp1989 commented 4 years ago

Hello @AliaksandrSiarohin !

I have tried the code using the following command python part_swap.py --source_image ./final.jpeg --target_video ./crop.mp4 --checkpoint vox-first-order.pth.tar --config ./motion-cosegmentation/config/vox-256-sem-15segments.yaml --supervised --first_order_motion_model --swap_index 1,2,3,4,5,6,7,8,9,10,11,12,13,14,15

How can I use the checkpoint of first_order_motion_model with changing the number of segments?

AliaksandrSiarohin commented 4 years ago

It is not possible, need retraining.

xupp1989 commented 4 years ago

Is the number of key point num_kp in first_order_motion_model correspond with the the number of segments?

AliaksandrSiarohin commented 4 years ago

Yes

xupp1989 commented 4 years ago

Thanks for the answer, makes sense now