Sync object-detection from metro-analytics-catalog

README.md

@@ -127,26 +127,20 @@ import cv2
 import numpy as np
 import openvino as ov
 
-COCO_NAMES = [
-    "person","bicycle","car","motorcycle","airplane","bus","train","truck",
-    "boat","traffic light","fire hydrant","stop sign","parking meter","bench",
-    "bird","cat","dog","horse","sheep","cow","elephant","bear","zebra",
-    "giraffe","backpack","umbrella","handbag","tie","suitcase","frisbee",
-    "skis","snowboard","sports ball","kite","baseball bat","baseball glove",
-    "skateboard","surfboard","tennis racket","bottle","wine glass","cup",
-    "fork","knife","spoon","bowl","banana","apple","sandwich","orange",
-    "broccoli","carrot","hot dog","pizza","donut","cake","chair","couch",
-    "potted plant","bed","dining table","toilet","tv","laptop","mouse",
-    "remote","keyboard","cell phone","microwave","oven","toaster","sink",
-    "refrigerator","book","clock","vase","scissors","teddy bear","hair drier",
-    "toothbrush",
-]
 CONF_THRESHOLD = 0.4
 INPUT_SIZE = 640
 
 core = ov.Core()
 model = core.read_model("yolo26n_openvino_model/yolo26n.xml")
 
+# YOLO26 embeds the 80 COCO class names in rt_info -- read them instead of
+# hardcoding the list. Ultralytics separates multi-word names with
+# underscores (e.g. "traffic_light"), so restore spaces for display.
+COCO_NAMES = [
+    name.replace("_", " ")
+    for name in model.get_rt_info()["model_info"]["labels"].value.split()
+]
+
 # Change device to "GPU" or "NPU" to run on integrated GPU or NPU.
 compiled = core.compile_model(model, "CPU")
 

export_and_quantize.sh

@@ -54,17 +54,6 @@ if [[ ! -f test.jpg ]]; then
 else
     echo "Already present: test.jpg"
 fi
-
-# Ask for approval before downloading models and sample files
-echo ""
-echo "This script will download:"
-echo "  - Model weights and/or sample files"
-echo ""
-read -p "Continue with downloads? (yes/no): " APPROVAL
-if [[ "${APPROVAL}" != "yes" ]]; then
-    echo "Download cancelled by user."
-    exit 0
-fi
 echo ""
 echo "--- Downloading sample test video ---"
 if [[ ! -f test_video.mp4 ]]; then