mmagic.models.editors.arcface

Package Contents

Classes

IDLossModel

Face id loss model.

class mmagic.models.editors.arcface.IDLossModel(ir_se50_weights=None)

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

extract_feats(x)

Extracting face features.

Parameters

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

Returns

Face features.

Return type

torch.Tensor

forward(pred=None, gt=None)

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)