Skip to content

Commit 8f55dfe

Browse files
authored
fix(wired_table_rec): Merge pull request #13 from Aues6uen11Z/main
Fix squeeze bug
2 parents d328ffc + 0d84311 commit 8f55dfe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

wired_table_rec/utils_table_recover.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def merge_adjacent_polys(polygons: np.ndarray) -> np.ndarray:
8585

8686
def combine_two_poly(polygons: np.ndarray, idxs: np.ndarray) -> np.ndarray:
8787
del_idxs, insert_boxes = [], []
88-
idxs = idxs.squeeze(0)
88+
idxs = idxs.squeeze(-1)
8989
for idx in idxs:
9090
# idx 和 idx + 1 是重合度过高的
9191
# 合并,取两者各个点的最大值

0 commit comments

Comments
 (0)