fix(wired_table_rec): Merge pull request #13 from Aues6uen11Z/main

Fix squeeze bug
RapidAI · Jun 14, 2024 · 8f55dfe · 8f55dfe
2 parents d328ffc + 0d84311
commit 8f55dfe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wired_table_rec/utils_table_recover.py b/wired_table_rec/utils_table_recover.py
@@ -85,7 +85,7 @@ def merge_adjacent_polys(polygons: np.ndarray) -> np.ndarray:
 
 def combine_two_poly(polygons: np.ndarray, idxs: np.ndarray) -> np.ndarray:
     del_idxs, insert_boxes = [], []
-    idxs = idxs.squeeze(0)
+    idxs = idxs.squeeze(-1)
     for idx in idxs:
         # idx 和 idx + 1 是重合度过高的
         # 合并，取两者各个点的最大值