LayoutLMv3_for_recepits2

Sleeping

mp-02 commited on Aug 8, 2024

Commit

4f4c048

verified ·

1 Parent(s): 10c2a49

Update sroie_inference.py

Files changed (1) hide show

sroie_inference.py CHANGED Viewed

@@ -88,7 +88,7 @@ def prediction(image):
     blur_boxes = []
     for prediction, box in zip(true_predictions, true_boxes):
-        if prediction != 'O' and prediction[2:] != 'TOTAL':
             blur_boxes.append(box)
     image = (blur(image, blur_boxes))

     blur_boxes = []
     for prediction, box in zip(true_predictions, true_boxes):
+        if prediction != 'O' and prediction != 'TOTAL':
             blur_boxes.append(box)
     image = (blur(image, blur_boxes))