Line 105 in train_AEI.py

taotaonice / FaceShifter

Try to reproduce FaceShifter

251 stars 62 forks source link

Line 105 in train_AEI.py #10

Open rookiecm opened 4 years ago

rookiecm commented 4 years ago

I am confused about line 105 in train_AEI.py which is same_person = diff_person. Could you please explain it ?

taotaonice commented 4 years ago

It is not a bug but my bad coding habit, I just want to set same_person and diff_person to all-one vector. I make a little change from the paper: I also define the L_rec on different person. I think this way may let Yt like Xs more in region that arcface interests in, Yt like Xt more in region that arcface does't care.