View source on GitHub
|
APIs to train face stylization model.
Modules
classification_ds module: Common classification dataset library.
constants module: Face stylizer model constants.
face_stylizer_options module: Options for building face stylizer.
hp module: Hyperparameters for training on-device face stylization models.
image_preprocessing module: ImageNet preprocessing.
loss_functions module: Loss function utility library.
model_opt module: Configurable model options for face stylizer models.
model_util module: Utilities for models.
ms module: Face stylizer model specification.
Classes
class FaceStylizer: FaceStylizer for building face stylization model.