fix for opencv changes

This commit is contained in:
Blake Blackshear 2022-02-21 06:32:17 -06:00
parent 20b4b503f0
commit cbf26e09a4

View File

@ -623,7 +623,8 @@ def process_frames(
idxs = cv2.dnn.NMSBoxes(boxes, confidences, 0.5, 0.4) idxs = cv2.dnn.NMSBoxes(boxes, confidences, 0.5, 0.4)
for index in idxs: 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): if clipped(obj, frame_shape):
box = obj[2] box = obj[2]
# calculate a new region that will hopefully get the entire object # calculate a new region that will hopefully get the entire object