search

ai-forever / ghost

A new one shot face swap approach for image and video domains
Apache License 2.0
1.25k stars 265 forks source link

how to set the "weight_eyes"? #17

Closed ruanjiyang closed 2 years ago

ruanjiyang commented 2 years ago

Dear sberbank-ai, I try to train on my datasets. and want to know how to set the weight_eyes? should I put 10 for weitght_eyes?

thanks!

AlexanderGroshev commented 2 years ago

@ruanjiyang you should use eye_detector_loss=True and set some value on weight_eyes. We recommend to set weitght_eyes around 800.

ruanjiyang commented 2 years ago

Dear AlexanderGroshev.

Thanks very much for your feedback. I will follow your suggestion.

The followings are the parameters which used for my training.

parser.add_argument('--weight_adv', default=1, type=float, help='Adversarial Loss weight') parser.add_argument('--weight_attr', default=10, type=float, help='Attributes weight') parser.add_argument('--weight_id', default=20, type=float, help='Identity Loss weight') parser.add_argument('--weight_rec', default=10, type=float, help='Reconstruction Loss weight') parser.add_argument('--weight_eyes', default=800., type=float, help='Eyes Loss weight') parser.add_argument('--eye_detector_loss', default=True, type=bool, help='If True eye loss with using AdaptiveWingLoss detector is applied to generator')