Update README.md

README.md

@@ -96,14 +96,14 @@ def detect_bboxes(masks: np.ndarray):
 
 bbox_pred = []
 for segment in segmentation:
-    bboxes, labels = [], []
+    boxes, labels = [], []
     for ii in range(1, len(model.config.label2id)):
         mm = segment == ii
         if mm.sum() > 0:
             bbx, lab = detect_bboxes(mm.numpy())
-            bboxes.extend(bbx)
+            boxes.extend(bbx)
             labels.extend([ii]*len(bbx))
-    bbox_pred.append(dict(bboxes=bboxes, labels=lables))
+    bbox_pred.append(dict(boxes=boxes, labels=lables))
 ```
 
 ### Citation