Segment and Label Videos Using Ultralytics Annotator / Image Segmentation, Pytorch Contents hide 1 Introduction 2 Ultralytics Annotator in a real segmentation workflow 3 Automatically Segment and Label Objects in Video with Python 3.1 Master Computer Vision 4 Automatically Segment Any Object in Video with YOLOv11 + Ultralytics Annotator 5 Set up a clean YOLOv11 environment for segmentation 6 Load the model and prepare your video input and output 7 Segment frame by frame and draw overlays with Ultralytics Annotator 8 Export the video, preview live, and shut everything down cleanly 9 FAQ : 9.1 What does Ultralytics Annotator do in this code? 9.2 Why do we check results[0].masks is not None? 9.3 What is stored in masks.xy? 9.4 Why do we use results[0] for a video frame? 9.5 How can I make segmentation faster? 9.6 Why does the output video sometimes fail to play? 9.7 How do I add confidence scores to labels? 9.8 Why does cap.read() return no frames? 9.9 Why are saved instance images full frames? 9.10 What is the easiest way to reduce mask flicker? 10 Conclusion 10.1 Connect : 10.2 Best AI Photo Tools (Backgrounds, Objects, Headshots) Last Updated on 22/04/2026 by Eran Feit Introduction Tip me and Download the code Ultralytics Annotator is a practical utility for turning raw model predictions into clear, human-readable visuals.In computer vision projects, predictions are only half the story.To really understand what a model is doing, you usually need to draw masks, labels, and colors on top of frames so you can verify quality quickly.That’s where Ultralytics Annotator fits in, especially when you’re working with segmentation outputs and want to see results instantly. In modern segmentation pipelines, you often deal with multiple instances per frame, each with its own polygon mask and class name.A clean annotation layer helps you confirm that masks are aligned, classes are correct, and object boundaries look reasonable.It also makes it much easier to communicate results to others, because a segmented overlay is immediately understandable, even without reading logs or digging into arrays.This is particularly useful when you process video frame-by-frame and need consistent visuals across thousands of frames. When paired with YOLO segmentation models, Ultralytics Annotator becomes part of a fast feedback loop.You run inference, overlay masks and labels, save outputs, and immediately spot issues like missed detections, swapped classes, or messy boundaries.Instead of guessing whether your segmentation looks good, you can visually validate it in a repeatable way.That visual validation is a big deal when you’re building reliable demos, tutorials, or real-world applications. In video segmentation workflows, annotation isn’t only for viewing results on-screen.It’s also about producing a polished output video that preserves the segmentation overlays and labels.That makes your final result shareable and measurable, and it helps you compare different models or settings over the same footage.The Annotator approach keeps the pipeline simple while still giving you professional-looking visualization. Ultralytics Annotator in a real segmentation workflow Ultralytics Annotator is designed to sit right inside your inference loop and handle the visualization layer without making the pipeline complicated.In a typical workflow, your model returns predicted classes and segmentation masks, and the annotator takes that information and renders it on the current frame.Instead of writing custom drawing code for every project, you get a consistent way to overlay masks, set colors, and place readable labels.This consistency matters when you want repeatable results across different videos and datasets. The main target in a segmentation video pipeline is speed and clarity at the same time.You want the overlay to be fast enough to run on every frame, but also clean enough to understand what the model predicted.With segmentation, the mask is often the most important output, because it shows the object’s exact shape instead of just a rectangle.A good annotator makes those masks visible, distinguishes overlapping objects, and keeps labels readable while the video is playing. At a high level, Ultralytics Annotator helps you answer the most important debugging questions quickly.Are masks snapped correctly to object boundaries.Are class names consistent with what you expect.Are there repeated false positives on similar background textures.When you can see masks and labels instantly, you can iterate on model choice, confidence thresholds, and input resolution much faster. In a complete “segment any object in video” script, the annotator also supports a polished output workflow.You can write the annotated frames into a new video file and optionally save per-instance images for inspection or dataset building.This turns a raw inference script into a usable tool for demos and experiments.The result is a practical pipeline where model predictions become visual outputs you can validate, share, and improve. Ultralytics annotator Automatically Segment and Label Objects in Video with Python This tutorial focuses on building a complete video-segmentation pipeline in Python, where every frame is processed, segmented, and annotated automatically.The core goal of the code is to take a raw video file, run a YOLOv11 segmentation model on each frame, and visually explain the results by drawing masks and class labels directly on the video.Instead of working with single images, the workflow is designed for continuous video streams, which is far more representative of real-world computer vision applications. At a high level, the code loads a pretrained YOLOv11 segmentation model and uses it to perform inference frame-by-frame.For each frame, the model predicts both object classes and precise segmentation masks.Those predictions are then passed to the Ultralytics Annotator, which overlays colored masks and readable labels on top of the original frame.This creates an immediate visual confirmation of what the model sees and how it understands the scene. A key target of this script is automation.Once the video capture starts, the entire process runs without manual intervention.Frames are read, segmented, annotated, displayed, and written to a new output video in a continuous loop.This makes the approach suitable for long videos, experiments, and demonstrations where consistency and repeatability matter. Beyond visualization, the code also supports saving segmented instances to disk.Each detected object can be stored as an image, organized by class name, which is useful for inspection, debugging, or building new datasets.By combining real-time visualization with structured output, the script turns segmentation inference into a practical tool rather than a one-off experiment. Link to the video tutorial here You can download the code here or here . TRY IT NOW Master Computer Vision Follow my latest tutorials and AI insights on my Personal Blog. Beginner Complete CV Bootcamp Foundation using PyTorch & TensorFlow. Get Started → Interactive Deep Learning with PyTorch Hands-on practice in an interactive environment. Start Learning → Advanced Modern CV: GPT & OpenCV4 Vision GPT and production-ready models. Go Advanced → Automatic object segmentation in video Automatically Segment Any Object in Video with YOLOv11 + Ultralytics Annotator Automatic video segmentation is one of the fastest ways to turn raw footage into structured visual data.Instead of drawing masks by hand, a segmentation model can generate pixel-accurate shapes for every object it detects in each frame.That makes it useful for dataset creation, video analytics, scene understanding, and content editing workflows. In this tutorial, the main focus is the Ultralytics Annotator.It gives you a clean way to draw colored mask overlays and readable labels directly on top of video frames.Once you combine it with a YOLOv11 segmentation model, you can build a full “read video → segment → overlay → export” pipeline in a compact Python script. The goal of the code is practical and production-like.You load a segmentation checkpoint, process the video frame by frame, visualize results live, and save the final segmented video to disk.You also export per-class instance images to folders, which is useful when you want training-ready outputs or quick inspection of what the model found. More segmentation tutorials that match this workflow How to segment multiple objects with YOLO Python This post uses a very similar YOLO segmentation flow and helps you understand masks, classes, and visualization at a comfortable pace. Boost Your Dataset With YOLOv8 Auto-Label Segmentation A great companion when you want to export masks and organize outputs into folders for dataset building and reuse. Fast YOLOv8 Dog Segmentation Tutorial for Video & Images Another end-to-end segmentation pipeline that focuses on real video and practical output saving. Set up a clean YOLOv11 environment for segmentation A stable environment is the difference between a smooth tutorial and hours of dependency issues.This setup creates a dedicated Conda environment, verifies your CUDA compiler, and installs a GPU-compatible PyTorch build.When you match PyTorch, CUDA, and Ultralytics versions, your segmentation inference will be faster and more predictable. The code below also pins Ultralytics and Pillow versions.Pinning is helpful when you want reproducible results across machines or when you are recording a tutorial and want zero surprises.After this step, you are ready to run YOLOv11 segmentation on video frames with OpenCV. ### Create a new Conda environment using Python 3.11 for this YOLOv11 project. conda create --name YoloV11-311 python=3.11 ### Activate the Conda environment so all installs stay isolated. conda activate YoloV11-311 ### Verify that CUDA is available by printing the nvcc compiler version. nvcc --version ### Install GPU-enabled PyTorch that matches CUDA 12.4 for faster inference. # Cuda 12.4 conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia ### Install the Ultralytics package version that includes YOLOv11 support. # install YoloV11 pip install ultralytics==8.3.59 ### Install Pillow for image saving and general image handling utilities. pip install pillow==11.1.0 Summary:You now have a dedicated environment that is ready for YOLOv11 segmentation work.If CUDA is installed correctly, inference will run significantly faster than CPU-only mode. Load the model and prepare your video input and output This section connects all the core building blocks.You import the required libraries, load the YOLOv11 segmentation checkpoint, and fetch the class names so labels stay readable.This is also where you define your input video path and your output directory. The VideoWriter configuration matters more than people expect.If you mismatch the resolution, codec, or FPS, you can get corrupted output files or videos that do not play smoothly.The code uses the input video’s width and height so the exported result stays consistent with the original footage. ### Import OpenCV so we can read video frames, display results, and write output video files. import cv2 ### Import YOLO from Ultralytics to load the segmentation model and run predictions. from ultralytics import YOLO ### Import Annotator and colors to draw masks and labels using the Ultralytics Annotator utilities. from ultralytics.utils.plotting import Annotator, colors ### Import os so we can create folders and build safe output paths. import os ### Load a custom YOLOv11 model trained for segmentation. model = YOLO("yolo11l-seg.pt") # Load a custom YOLOv11 model trained for segmentation ### Read the class-name mapping from the loaded model so labels match the model training. names = model.model.names # Get the class names from the model ### Print class names once so you can confirm what the model is able to detect. print("Class Names:", names) ### Open the input video file using OpenCV VideoCapture. cap = cv2.VideoCapture("Best-Semantic-Segmentation-models/Yolo-V11/Auto segment any Object/town.mp4") ### Choose the output folder where the segmented video and instance frames will be saved. output_folder = "d:/temp/output_segment_yoloV11" ### Create the output folder if it does not exist so saving never fails later. os.makedirs(output_folder, exist_ok=True) # Create output folder if it doesn't exist ### Create a VideoWriter to export the final segmented video with the same resolution as the input video. out = cv2.VideoWriter(os.path.join(output_folder, "output_segment_yoloV11.mp4"), cv2.VideoWriter_fourcc(*'MJPG'), 30 , (int(cap.get(3)), int(cap.get(4)))) # Define the codec and create VideoWriter object Summary:You have a loaded YOLOv11 segmentation model and a working video I/O pipeline.At this point, the only missing piece is the per-frame segmentation loop and visualization. Useful video-processing patterns to reuse Real-time Youtube video stream extraction and object detection Shows a frame-by-frame pipeline on streaming video, which is a great extension after you master local MP4 processing. Object Detection Heatmap For Tracking Moving Dogs Turns frame-level detections into a higher-level visualization, which pairs nicely with segmentation outputs for analytics. OpenCV Car Detection : Background Subtraction Tutorial A classic OpenCV video pipeline that helps you understand frame handling, masks, and visualization even without deep learning. Segment frame by frame and draw overlays with Ultralytics Annotator This is the heart of the tutorial.You read frames in a loop, send each frame into the model, and then extract masks and classes from the results.When masks exist, you use the Ultralytics Annotator to draw segmentation overlays and labels right on the frame. A useful mental model is to treat each frame as a standalone image inference.YOLOv11 returns a results list, and for a single frame you usually access results[0].From there, you check whether masks exist, pull polygon coordinates, and render them with consistent colors per class. ### Start an infinite loop to process the video frame by frame until the video ends or the user quits. while True: ### Read one frame from the input video. ret , im0 = cap.read() # Read a frame from the video ### If OpenCV fails to read a frame, the video is over or the path is invalid, so we exit cleanly. if not ret: print("No frame read from video. Exiting...") break # Break the loop if no frame is read ### Run YOLOv11 segmentation on the current frame. results = model.predict(im0) ### Continue only if the model produced segmentation masks for this frame. if results[0].masks is not None: ### Extract class IDs for each detected instance and move them to CPU for easy handling. clss = results[0].boxes.cls.cpu().tolist() # ### Extract polygon mask coordinates for each instance. masks = results[0].masks.xy ### Create an Ultralytics Annotator so we can draw masks and labels onto the frame. annotator = Annotator(im0, line_width=2) ### Loop over each predicted instance mask and its class ID. for idx , (mask, cls) in enumerate(zip(masks, clss)): ### Convert the class ID into a human-readable label using the model's class name map. det_label = names[int(cls)] # Get the class name for the detected object #print(int(cls)) ### Draw the segmentation mask overlay and the label using a consistent class-based color. annotator.seg_bbox(mask = mask , mask_color=colors(int(cls), True), label = det_label) ### Save each instance segmented object instance_folder = os.path.join(output_folder, det_label) os.makedirs(instance_folder, exist_ok=True) instance_path = os.path.join(instance_folder, f"{det_label}_{idx}.png") cv2.imwrite(instance_path, im0) Summary:Your frames are now being segmented and annotated in real time using Ultralytics Annotator.Each detected object gets a mask overlay, a readable label, and an exported frame saved into a class-based folder. Export the video, preview live, and shut everything down cleanly After segmentation and overlays, you still need two practical features.You want a live preview window so you can validate results while the script runs.You also want an exported video file so you can share results or reuse them in later steps. This final part writes each annotated frame to the VideoWriter and shows it with OpenCV.It also adds a keyboard exit so you can stop early without killing the Python process.Finally, it releases resources cleanly so the output video is properly finalized and not corrupted. ### Display the annotated frame in a live OpenCV window so you can inspect segmentation quality. cv2.imshow("Result", im0) # Display the result ### Write the annotated frame into the output video so the final MP4 contains the overlays. out.write(im0) # Write the frame to the output video ### Allow quitting the loop by pressing the 'q' key while the window is active. if cv2.waitKey(1) & 0xFF == ord('q'): break ### Release the input video so the file handle closes properly. cap.release() # Release the video capture object ### Release the writer so OpenCV finalizes and saves the output video correctly. out.release() # Release the video writer object ### Close all OpenCV windows to end the session cleanly. cv2.destroyAllWindows() # Close all OpenCV windows Summary:You now have a full pipeline that previews segmentation live and exports the final segmented video.The script exits safely and releases resources so your output file is saved correctly. Good next steps after YOLO video segmentation Segment Anything Tutorial: Fast Auto Masks in Python Shows a fast auto-mask workflow that complements YOLO segmentation when you want flexible prompt-based masking. One-Click Segment Anything in Python (SAM ViT-H) Great when you want interactive segmentation, and a different way to think about masks compared to model-only inference. Detectron2 Panoptic Segmentation Made Easy for Beginners Helps you compare instance masks to full-scene segmentation concepts when you want richer outputs than instance-only masks. FAQ : What does Ultralytics Annotator do in this code? It draws segmentation overlays and labels on the frame. This gives you clean visualization without manual polygon drawing code. Why do we check results[0].masks is not None? Some frames have no detections, so there are no masks to draw. The check prevents errors and keeps the loop stable. What is stored in masks.xy? It contains polygon points outlining each instance mask. These polygons are convenient for fast drawing and lightweight storage. Why do we use results[0] for a video frame? Ultralytics returns a list of results, one per input image. With one frame input, the first item contains the prediction data. How can I make segmentation faster? Use a smaller model, resize frames, and run on a GPU. You can also process every Nth frame if you do not need every frame. Why does the output video sometimes fail to play? Codec and resolution mismatches are common causes. Ensure the VideoWriter resolution matches the frames you write. How do I add confidence scores to labels? Read confidence values from the result boxes and format them into the label string. This is helpful for debugging detection quality. Why does cap.read() return no frames? It usually means the video ended, the path is wrong, or the codec is unsupported. Verifying the path and file playback is a quick check. Why are saved instance images full frames? The script writes the annotated frame for each instance index. For true object crops, you would mask and crop per instance. What is the easiest way to reduce mask flicker? Increase confidence thresholds and reduce input blur by resizing thoughtfully. For stronger stability, add lightweight tracking across frames. Conclusion This tutorial demonstrates a complete and practical segmentation pipeline for video.You start with a clean environment, load a YOLOv11 segmentation checkpoint, and build a reliable OpenCV loop for reading frames.Once predictions are available, the Ultralytics Annotator makes it simple to turn raw masks into a viewer-friendly overlay with labels. The output structure is also intentionally useful.You get a final segmented video that is easy to share, review, and archive.You also get class-organized exported frames, which helps you debug model behavior and can act as a starting point for dataset creation. From here, the next natural upgrades are straightforward.You can add confidence thresholds, export true per-instance crops, or introduce object tracking to keep instance IDs consistent across frames.But even without those extras, this script already covers the core workflow most real segmentation video projects need. Connect : ☕ Buy me a coffee — https://ko-fi.com/eranfeit 🖥️ Email : feitgemel@gmail.com 🌐 https://eranfeit.net 🤝 Fiverr : https://www.fiverr.com/s/mB3Pbb Enjoy, Eran Best AI Photo Tools (Backgrounds, Objects, Headshots) Over 30 AI-powered tools in one place for editing and creating photos. Create professional photos using AI-powered background removal and retouching. Your personal AI photographer for headshots and profile images.
Skip to content