Module: mediapipe_model_maker.face_stylizer.face_stylizer.loss_functions

Loss function utility library.

Modules

file_util module: Utilities for files.

model_util module: Utilities for models.

Classes

class FocalLoss: Implementation of focal loss (https://arxiv.org/pdf/1708.02002.pdf).

class ImagePerceptualQualityLoss: Image perceptual quality loss.

class PerceptualLoss: Base class for perceptual loss model.

class PerceptualLossWeight: The weight for each perceptual loss.

class SparseFocalLoss: Sparse implementation of Focal Loss.

class VGGPerceptualLoss: Perceptual loss based on VGG19 pretrained on the ImageNet dataset.