Fix Bug 2

counter_live_rknn_bytetrack.py

@@ -985,12 +985,8 @@ def run():
         score_sigmoid=SCORE_SIGMOID,
     )
 
-    CLASS_IDS = {
-        os.getenv("CLASS_AYAM", "ayam"): 0,
-        os.getenv("CLASS_TALENAN", "talenan"): 1,
-    }
-    ayam_cls = CLASS_IDS[CLASS_AYAM]
-    talenan_cls = CLASS_IDS[CLASS_TALENAN]
+    ayam_cls = int(os.getenv("AYAM_CLASS_ID", "0"))
+    talenan_cls = int(os.getenv("TALENAN_CLASS_ID", "1"))
 
     ayam_tracker = ByteTracker(
         track_high_thresh=TRACK_HIGH_THRESH_0,
@@ -1115,6 +1111,40 @@ def run():
                 talenan_tracker.update(talenan_boxes_xyxy, talenan_scores)
             )
 
+            if os.getenv("DEBUG_TRACKING", "").lower() == "true":
+                if len(ayam_boxes_xyxy) > 0 or len(talenan_boxes_xyxy) > 0:
+                    ayam_scores_str = (
+                        f"  ayam scores: {ayam_scores.round(3).tolist()}"
+                        if len(ayam_scores) > 0
+                        else ""
+                    )
+                    talenan_scores_str = (
+                        f"  talenan scores: {talenan_scores.round(3).tolist()}"
+                        if len(talenan_scores) > 0
+                        else ""
+                    )
+                    ayam_tracks_str = (
+                        f"  ayam det→track: {dict(ayam_det_to_track)}"
+                        if ayam_det_to_track
+                        else ""
+                    )
+                    talenan_tracks_str = (
+                        f"  talenan det→track: {dict(talenan_det_to_track)}"
+                        if talenan_det_to_track
+                        else ""
+                    )
+                    crossing_str = (
+                        f"  ayam_line_crossed: {sorted(ayam_line_crossed)}"
+                        if ayam_line_crossed
+                        else ""
+                    )
+                    print(
+                        f"[DEBUG F{frame_idx}] ayam_dets={len(ayam_boxes_xyxy)} "
+                        f"talenan_dets={len(talenan_boxes_xyxy)} "
+                        f"line_x={line_x}{ayam_scores_str}{talenan_scores_str}"
+                        f"{ayam_tracks_str}{talenan_tracks_str}{crossing_str}"
+                    )
+
             # Process talenan crossings
             for di in range(len(talenan_boxes_xyxy)):
                 tid = talenan_det_to_track.get(di)