File size: 520 Bytes
c446951 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | import cv2
import inference
import supervision as sv
annotator = sv.BoxAnnotator()
def render(predictions, image):
print(predictions)
image = annotator.annotate(
scene=image, detections=sv.Detections.from_roboflow(predictions)
)
cv2.imshow("Prediction", image)
cv2.waitKey(1)
# replace with your own RTSP stream
inference.Stream(
source="rtsp://127.0.0.1:8000/test",
model="microsoft-coco/9",
output_channel_order="BGR",
use_main_thread=True,
on_prediction=render,
)
|