What this visualizer does
It classifies every mask pixel, then renders a colored overlay, category mask, or foreground cutout with a replaced background.
Computer vision visualizer
Separate people, hair, skin, clothes, accessories, and background pixels directly in your browser with MediaPipe.
The multiclass selfie model runs locally and returns a category for every mask pixel.
It classifies every mask pixel, then renders a colored overlay, category mask, or foreground cutout with a replaced background.
import mediapipe as mp
BaseOptions = mp.tasks.BaseOptions
ImageSegmenter = mp.tasks.vision.ImageSegmenter
ImageSegmenterOptions = mp.tasks.vision.ImageSegmenterOptions
options = ImageSegmenterOptions(
base_options=BaseOptions(model_asset_path="selfie_multiclass.tflite"),
output_category_mask=True
)
with ImageSegmenter.create_from_options(options) as segmenter:
image = mp.Image.create_from_file("portrait.jpg")
result = segmenter.segment(image)
category_mask = result.category_mask.numpy_view()Overlay shows classes over the image. Mask shows only categories. Background keeps the person and replaces or blurs class zero.
Portrait framing and even lighting produce cleaner edges. Fine hair and transparent objects remain challenging for compact models.