Filtering With LineZoneAnnotator

[mwitiderrick]

Is filtering supported with LineZoneAnnotator?

Getting TypeError: unsupported operand type(s) for &: 'bool' and 'NoneType'

import numpy as np
import supervision as sv
from deepsparse import Pipeline

STORE_VIDEO_PATH = "retail.mp4"

start = sv.Point(x = 1000, y = 900)
end =  sv.Point(x = 200, y = 500)

video_info = sv.VideoInfo.from_video_path(STORE_VIDEO_PATH)
zone = sv.LineZone(start=start,end=end)

# extract video frame
generator = sv.get_video_frames_generator(STORE_VIDEO_PATH)
iterator = iter(generator)
frame = next(iterator)

# detect
model_stub = "zoo:cv/detection/yolov5-l6/pytorch/ultralytics/coco/base-none"
yolo_pipeline = Pipeline.create(task="yolo",model_path=model_stub)
pipeline_results = yolo_pipeline(images=frame, iou_thres=0.6, conf_thres=0.001)
detections = sv.Detections.from_deepsparse(pipeline_results)
mask = zone.trigger(detections=detections)
detections = detections[(detections.class_id == 0) & (detections.confidence > 0.5) & mask]


line_annotator = sv.LineZoneAnnotator(thickness=4, text_thickness=4, text_scale=2)
frame = line_annotator.annotate(frame=frame, line_counter=zone)

%matplotlib inline
sv.plot_image(frame, (16, 16))