YuvalNirkin / face_swap

End-to-end, automatic face swapping pipeline
GNU General Public License v3.0
825 stars 203 forks source link

how to train face segmentation model #32

Closed rootkit closed 5 years ago

rootkit commented 5 years ago

Hi YuvalNiirkin I wanna train face segmentation model using my dataset, what should I do?

Thanks

John

YuvalNirkin commented 5 years ago

Use the following repository to train a Caffe FCN model: https://github.com/shelhamer/fcn.berkeleyvision.org

Yuval

rootkit commented 5 years ago

Thanks