fix for opencv changes

2026-07-20 10:49:00 +03:00 · 2022-02-21 06:44:00 -06:00
parent 6f85202998
commit ce4550c430
1 changed files with 2 additions and 1 deletions
@@ -623,7 +623,8 @@ def process_frames(
                    idxs = cv2.dnn.NMSBoxes(boxes, confidences, 0.5, 0.4)

                    for index in idxs:
-                        obj = group[index[0]]
+                        index = index if isinstance(index, np.int32) else index[0]
+                        obj = group[index]
                        if clipped(obj, frame_shape):
                            box = obj[2]
                            # calculate a new region that will hopefully get the entire object