Fix multiple object detection

app.py

@@ -73,11 +73,11 @@ def draw_boxes(image, boxes, class_names, scores, max_boxes=10, min_score=0.1):
         print("Font not found, using default font.")
         font = ImageFont.load_default()
 
-    for i in range(min(boxes.shape[0], max_boxes)):
-        if scores[i][0] >= min_score:
-            ymin, xmin, ymax, xmax = tuple(boxes[i][0])
+    for i in range(min(boxes.shape[1], max_boxes)):
+        if scores[0][i] >= min_score:
+            ymin, xmin, ymax, xmax = tuple(boxes[0][i])
             display_str = "{}: {}%".format(class_names[i],
-                                           int(100 * scores[i][0]))
+                                           int(100 * scores[0][i]))
             color = colors[hash(class_names[i]) % len(colors)]
             image_pil = Image.fromarray(np.uint8(image)).convert("RGB")
             draw_bounding_box_on_image(
@@ -106,9 +106,7 @@ def run_detector(url_input, image_input, minscore=0.1):
 
     result = {key: value.numpy() for key, value in result.items()}
 
-    print("Found %d objects." % len(result["detection_scores"]))
-    labels = ["cyclist" for _ in range(len(result["detection_scores"]))]
-    print(labels)
+    labels = ["cyclist" for _ in range(len(result["detection_scores"][0]))]
 
     image_with_boxes = draw_boxes(
         np.array(img), result["detection_boxes"],