Table of Contents
Shortcuts

mmagic.models.editors.arcface.id_loss

Module Contents

Classes

IDLossModel

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://gg0ltg.by.files.1drv.com/y4m3fNNszG03z9n8JQ7EhdtQKW8tQVQMFBisPVRgoXi_UfP8pKSSqv8RJNmHy2Ja...'[源代码]
extract_feats(x)[源代码]

Extracting face features.

参数

x (torch.Tensor) – Image tensor of faces.

返回

Face features.

返回类型

torch.Tensor

forward(pred=None, gt=None)[源代码]

Calculate face loss.

参数

  • pred (torch.Tensor, optional) – Predictions of face images. Defaults to None.

  • gt (torch.Tensor, optional) – Ground truth of face images. Defaults to None.

返回

A tuple contain face similarity loss and

improvement.

返回类型

Tuple(float, float)