mmagic.models.editors.arcface.id_loss
¶
Module Contents¶
Classes¶
Face id loss model. |
- class mmagic.models.editors.arcface.id_loss.IDLossModel(ir_se50_weights=None)[源代码]¶
Bases:
torch.nn.Module
Face id loss model.
- 参数
ir_se50_weights (str, optional) – Url of ir-se50 weights. Defaults to None.
- _ir_se50_url = 'https://download.openxlab.org.cn/models/rangoliu/Arcface-IR-SE50/weight/Arcface-IR-SE50'[源代码]¶