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

[xupp1989]

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]

It is not possible, need retraining.

[xupp1989]

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

[AliaksandrSiarohin]

Yes

[xupp1989]

Thanks for the answer, makes sense now