cmarkea
/

dit-base-layout-detection

Image Segmentation

Model card Files Files and versions

Cyrile commited on Sep 11, 2024

Commit

01a8d67

·

verified ·

1 Parent(s): 9847b1d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -103,7 +103,7 @@ for segment in segmentation:
     for ii in range(1, len(model.config.label2id)):
         mm = segment == ii
         if mm.sum() > 0:
-            bbx, lab = detect_bboxes(mm.numpy())
             boxes.extend(bbx)
             labels.extend([ii]*len(bbx))
     bbox_pred.append(dict(boxes=boxes, labels=lables))

     for ii in range(1, len(model.config.label2id)):
         mm = segment == ii
         if mm.sum() > 0:
+            bbx = detect_bboxes(mm.numpy())
             boxes.extend(bbx)
             labels.extend([ii]*len(bbx))
     bbox_pred.append(dict(boxes=boxes, labels=lables))