Przeglądaj źródła

Avoid cropping empty images.

jinghuan-Chen 2 miesięcy temu
rodzic
commit
8bb8b715c1
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      mineru/model/table/rec/unet_table/main.py

+ 1 - 1
mineru/model/table/rec/unet_table/main.py

@@ -183,7 +183,7 @@ class WiredTableRecognition:
                 continue
                 continue
             # 从img中截取对应的区域
             # 从img中截取对应的区域
             x1, y1, x2, y2 = int(box[0][0])+1, int(box[0][1])+1, int(box[2][0])-1, int(box[2][1])-1
             x1, y1, x2, y2 = int(box[0][0])+1, int(box[0][1])+1, int(box[2][0])-1, int(box[2][1])-1
-            if x1 >= x2 or y1 >= y2:
+            if x1 >= x2 or y1 >= y2 or x1 < 0 or y1 < 0:
                 # logger.warning(f"Invalid box coordinates: {x1, y1, x2, y2}")
                 # logger.warning(f"Invalid box coordinates: {x1, y1, x2, y2}")
                 continue
                 continue
             # 判断长宽比
             # 判断长宽比