La tâche MediaPipe Face Sylizer vous permet d'appliquer des styles aux visages sur une image. Vous pouvez utiliser cette tâche pour créer des avatars virtuels de différents styles.
L'exemple de code décrit dans ces instructions est disponible sur GitHub. Pour en savoir plus sur les fonctionnalités, les modèles et les options de configuration de cette tâche, consultez la présentation.
Exemple de code
L'exemple de code MediaPipe Tasks est une implémentation de base d'une application Face Stylizer pour Android. L'exemple applique un style de visage aux images fournies à l'application.
Vous pouvez utiliser l'application comme point de départ pour votre propre application Android ou vous y référer lorsque vous modifiez une application existante. L'exemple de code Face Stylizer est hébergé sur GitHub.
Télécharger le code
Les instructions suivantes vous expliquent comment créer une copie locale de l'exemple de code à l'aide de l'outil de ligne de commande git.
Pour télécharger l'exemple de code, procédez comme suit:
- Clonez le dépôt git à l'aide de la commande suivante :
git clone https://github.com/google-ai-edge/mediapipe-samples
- Si vous le souhaitez, vous pouvez configurer votre instance Git pour utiliser le paiement creux afin de n'avoir que les fichiers de l'application exemple Face Stylizer :
cd mediapipe git sparse-checkout init --cone git sparse-checkout set examples/face_stylization/android
Après avoir créé une version locale de l'exemple de code, vous pouvez importer le projet dans Android Studio et exécuter l'application. Pour obtenir des instructions, consultez le guide de configuration pour Android.
Composants clés
Les fichiers suivants contiennent le code crucial pour cet exemple d'application de stylisation de visage:
- FaceStylizationHelper.kt : initialise le style de visage, et gère la sélection du modèle et du délégué.
- MainActivity.kt : fournit les résultats et les résultats, et gère les erreurs.
Préparation
Cette section décrit les étapes clés de la configuration de votre environnement de développement et de vos projets de code spécifiquement pour l'utilisation de Face Stylizer. Pour obtenir des informations générales sur la configuration de votre environnement de développement pour l'utilisation des tâches MediaPipe, y compris sur les exigences de version de la plate-forme, consultez le guide de configuration pour Android.
Dépendances
La tâche "Face Stylizer" utilise la bibliothèque com.google.mediapipe:tasks-vision. Ajoutez cette dépendance au fichier build.gradle de votre application Android:
dependencies {
implementation 'com.google.mediapipe:tasks-vision:latest.release'
}
Modèle
La tâche MediaPipe Face Sylizer nécessite un groupe de modèles entraînés compatibles avec cette tâche. Pour en savoir plus sur les modèles entraînés disponibles pour le styliste de visage, consultez la section Modèles de la présentation des tâches.
Sélectionnez et téléchargez le modèle, puis stockez-le dans le répertoire de votre projet:
<dev-project-root>/src/main/assets
Spécifiez le chemin d'accès du modèle dans le paramètre ModelAssetPath.
val modelName = "https://storage.googleapis.com/mediapipe-models/face_stylizer/blaze_face_stylizer/float32/latest/face_stylizer_color_sketch.task"
baseOptionsBuilder.setModelAssetPath(modelName)
Créer la tâche
La tâche MediaPipe Face Sylizer utilise la fonction createFromOptions() pour configurer la tâche. La fonction createFromOptions() accepte des valeurs pour les options de configuration. Pour en savoir plus sur les options de configuration, consultez la section Options de configuration.
val baseOptionsBuilder = BaseOptions.builder().setModelAssetPath(modelName)
val baseOptions = baseOptionBuilder.build()
val optionsBuilder =
FaceStylizer.FaceStylizerOptions.builder()
.setBaseOptions(baseOptionsBuilder.build())
val options = optionsBuilder.build()
FaceStylizer =
FaceStylizer.createFromOptions(context, options)
Options de configuration
Cette tâche dispose des options de configuration suivantes pour les applications Android:
| Nom de l'option | Description | Plage de valeurs | Valeur par défaut |
|---|---|---|---|
errorListener |
Définit un écouteur d'erreurs facultatif. | N/A |
Not set |
Préparation des données
L'outil de retouche des visages fonctionne avec des images fixes. Cette tâche gère le prétraitement de l'entrée des données, y compris le redimensionnement, la rotation et la normalisation de la valeur. Le code suivant montre comment transférer des données pour traitement.
import com.google.mediapipe.framework.image.BitmapImageBuilder
import com.google.mediapipe.framework.image.MPImage
// Convert the input Bitmap object to an MPImage object to run inference
val mpImage = BitmapImageBuilder(image).build()
Exécuter la tâche
Utilisez la méthode FaceStylizer.stylize() sur l'image d'entrée pour exécuter le stylet:
val result = FaceStylizer.stylize(mpImage)
Gérer et afficher les résultats
Face Stylizer renvoie un objet FaceStylizerResult, qui contient un objet MPImage avec un style du visage le plus proéminent dans l'image d'entrée.
Voici un exemple des données de sortie de cette tâche:
La sortie ci-dessus a été créée en appliquant le modèle Color sketch à l'image d'entrée suivante:
