diff --git a/pycoral/adapters/detect.py b/pycoral/adapters/detect.py index f7b63cd..e1db1ee 100644 --- a/pycoral/adapters/detect.py +++ b/pycoral/adapters/detect.py @@ -234,4 +234,4 @@ def make(i): bbox=BBox(xmin=xmin, ymin=ymin, xmax=xmax, ymax=ymax).scale(sx, sy).map(int)) - return [make(i) for i in range(count) if scores[i] >= score_threshold] + return [make(i) for i in range(len(scores)) if scores[i] >= score_threshold]