Any suggestion for improve the face_swap precision? How to tune the parameter?

YuvalNirkin / face_swap

End-to-end, automatic face swapping pipeline

GNU General Public License v3.0

825 stars 203 forks source link

Any suggestion for improve the face_swap precision? How to tune the parameter? #46

Closed wackxu closed 5 years ago

wackxu commented 5 years ago

@YuvalNirkin Could you give me some suggestions?

YuvalNirkin commented 5 years ago

For best performance use face_swap_single2many or copy its pipeline. The following parameters are most significant:

seg_model = ../data/face_seg_fcn8s_300.caffemodel               # try the 500x500 model
seg_deploy = ../data/face_seg_fcn8s_300_deploy.prototxt         # try the 500x500 model
generic = 0                                  # use generic mode (disable shape regression)
source_max_res = 500                         # sources max resolution (try other resolutions)
target_max_res = 500                         # targets max resolution (try other resolutions)

wackxu commented 4 years ago

@YuvalNirkin If we use our data to retrain the model, will there be a significant improvement in performance?