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)[source]¶
Bases:
torch.nn.Module
Face id loss model.
- Parameters
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[source]¶
- extract_feats(x)[source]¶
Extracting face features.
- Parameters
x (torch.Tensor) – Image tensor of faces.
- Returns
Face features.
- Return type
torch.Tensor
- forward(pred=None, gt=None)[source]¶
Calculate face loss.
- Parameters
pred (torch.Tensor, optional) – Predictions of face images. Defaults to None.
gt (torch.Tensor, optional) – Ground truth of face images. Defaults to None.
- Returns
- A tuple contain face similarity loss and
improvement.
- Return type
Tuple(float, float)