I want to show zone_annotator all the time

[REZIZ-TER]

Search before asking

• I have searched the Supervision issues and found no similar feature requests.

Question

Now I have a problem. zone_annotator It is displayed only when an object is detected. But if there is no object in zone_annotator It'll disappear but that's not what I want. I want it to display all the time.

import numpy as np
import supervision as sv
from ultralytics import YOLO
import argparse
import numpy as np
from inference.models.utils import get_roboflow_model


model_path = "D:\\Private\\Y3Project\\python_project\\Weights\\w2024-04-27\\best.pt"
model = YOLO(model_path)
tracker = sv.ByteTrack()
box_annotator = sv.BoundingBoxAnnotator()
label_annotator = sv.LabelAnnotator()

ZONE_POLYGON = np.array([
    [0.1, 0.1],
    [0.9, 0.1],
    [0.9, 0.9],
    [0.1, 0.9],
    [0.1, 0.1]
])

def parse_arguments() -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="YOLOv8 live")
    parser.add_argument(
        "--webcam-resolution",
        default=[1280, 720],
        nargs=2,
        type=int
    )
    args = parser.parse_args()
    return args

args = parse_arguments()
zone_polygon = (ZONE_POLYGON * np.array(args.webcam_resolution)).astype(int)

zone = sv.PolygonZone(polygon=zone_polygon, frame_resolution_wh=tuple(args.webcam_resolution))
zone_annotator = sv.PolygonZoneAnnotator(
    zone=zone,
    color=sv.Color.RED,
    text_color=sv.Color.WHITE,
    thickness=1,
    text_thickness=1,
    text_scale=1
)

def callback(frame: np.ndarray, _: int) -> np.ndarray:

    results = model(frame)[0]
    detections = sv.Detections.from_ultralytics(results)
    detections = tracker.update_with_detections(detections)

    if len(detections) > 0 and len(detections.class_id) == len(detections.tracker_id):
        labels = [
            f"#{tracker_id} {results.names[class_id]}"
            for class_id, tracker_id
            in zip(detections.class_id, detections.tracker_id)
            ]

        annotated_frame = box_annotator.annotate(frame.copy(), detections=detections)
        annotated_frame = zone_annotator.annotate(annotated_frame.copy())
        label_annotators = label_annotator.annotate(annotated_frame.copy(), detections=detections, labels=labels)
        mask = zone.trigger(detections=detections)
        return label_annotators
        
    else:
        return frame
    
myvid="D:\\Private\Y3Project\\Filter_Pic\\7755db6e-8cba-4e35-a645-621b2e813b60.mp4"
sv.process_video(
    source_path=myvid4,
    target_path="D:\\Private\\Y3Project\\python_project\\runs\\myresult\\myvid.mp4",
    callback=callback
)

Additional

No response

[LinasKo]

Hi @REZIZ-TER 👋

This is because you're checking for len(detections) > 0 before annotating.

If you moved annotated_frame = zone_annotator.annotate(annotated_frame.copy()) outside the if, for example - right below detections = tracker.update_with_detections(detections), you'd get the zone annotations every time.