Detyra MediaPipe Face Stylizer ju lejon të aplikoni stilizime të fytyrës tek fytyrat në një imazh. Ju mund ta përdorni këtë detyrë për të krijuar avatarë virtualë në stile të ndryshme.
Shembulli i kodit i përshkruar në këto udhëzime është i disponueshëm në GitHub . Për më shumë informacion rreth aftësive, modeleve dhe opsioneve të konfigurimit të kësaj detyre, shihni Përmbledhjen .
Shembull kodi
Kodi shembull për Face Stylizer ofron një zbatim të plotë të kësaj detyre në Python për referencën tuaj. Ky kod ju ndihmon të testoni këtë detyrë dhe të filloni ndërtimin e stilizuesit tuaj të fytyrës. Mund të shikoni, ekzekutoni dhe modifikoni kodin shembull të Face Stylizer duke përdorur vetëm shfletuesin tuaj të internetit.
Konfigurimi
Ky seksion përshkruan hapat kryesorë për konfigurimin e mjedisit tuaj të zhvillimit dhe projekteve të kodit në mënyrë specifike për të përdorur Face Stylizer. Për informacion të përgjithshëm mbi konfigurimin e mjedisit tuaj të zhvillimit për përdorimin e detyrave të MediaPipe, duke përfshirë kërkesat e versionit të platformës, shihni udhëzuesin e konfigurimit për Python .
Paketat
Detyra MediaPipe Face Stylizer kërkon paketën mediapipe PyPI. Ju mund t'i instaloni dhe importoni këto varësi me sa vijon:
$ python -m pip install mediapipe
Importet
Importoni klasat e mëposhtme për të hyrë në funksionet e detyrës Face Stylizer:
import mediapipe as mp
from mediapipe.tasks import python
from mediapipe.tasks.python import vision
Model
Detyra MediaPipe Face Stylizer kërkon një model të trajnuar që është në përputhje me këtë detyrë. Për më shumë informacion mbi modelet e disponueshme të trajnuara për Face Stylizer, shihni seksionin Modelet e përmbledhjes së detyrave.
Zgjidhni dhe shkarkoni modelin dhe më pas ruajeni në një direktori lokale:
model_path = '/absolute/path/to/face_stylizer.task'
Përdorni parametrin e objektit BaseOptions model_asset_path për të specifikuar shtegun e modelit që do të përdoret. Për një shembull kodi, shihni seksionin tjetër.
Krijo detyrën
Detyra MediaPipe Face Stylizer përdor funksionin create_from_options për të konfiguruar detyrën. Funksioni create_from_options pranon vlera për opsionet e konfigurimit që duhen trajtuar.
Kodi i mëposhtëm tregon se si të ndërtohet dhe konfigurohet kjo detyrë.
import mediapipe as mp
BaseOptions = mp.tasks.BaseOptions
Facestylizer = mp.tasks.vision.face_stylizer
FacestylizerOptions = mp.tasks.vision.FaceStylizerOptions
# Create a face stylizer instance with the image mode:
options = FacestylizerOptions(
base_options=BaseOptions(model_asset_path=model_path),
with Facestylizer.create_from_options(options) as stylizer:
# The stylizer is initialized. Use it here.
# ...
Përgatitni të dhënat
Përgatitni hyrjen tuaj si një skedar imazhi ose një grup numpy, më pas konvertojeni atë në një objekt mediapipe.Image . Nëse hyrja juaj është një skedar video ose transmetim i drejtpërdrejtë nga një kamerë në internet, mund të përdorni një bibliotekë të jashtme si OpenCV për të ngarkuar kornizat tuaja të hyrjes si grupe të pakëndshme.
import mediapipe as mp
# Load the input image from an image file.
mp_image = mp.Image.create_from_file('/path/to/image')
# Load the input image from a numpy array.
mp_image = mp.Image(image_format=mp.ImageFormat.SRGB, data=numpy_image)
Drejtoni detyrën
Stilizuesi i fytyrës përdor funksionin stylize për të nxjerrë përfundime. Për stilizimin e fytyrës, kjo përfshin parapërpunimin e të dhënave hyrëse dhe stilizimin e fytyrave në imazh.
Kodi i mëposhtëm tregon se si të ekzekutohet përpunimi me modelin e detyrës.
# Perform face stylization on the provided single image.
# The face stylizer must be created with the image mode.
face_stylizer_result = stylizer.stylize(mp_image)
Trajtoni dhe shfaqni rezultatet
Stilizuesi i fytyrës kthen një objekt Image me një stilizim të fytyrës më të spikatur brenda imazhit të hyrjes.
Më poshtë tregon një shembull të të dhënave dalëse nga kjo detyrë:
Dalja e mësipërme u krijua duke aplikuar modelin e skicës me ngjyra në imazhin e mëposhtëm hyrës:
