From 0d843113fcfdede2a88a752ad4cd2e43ffb388a7 Mon Sep 17 00:00:00 2001
From: Aues6uen11Z <79563000+Aues6uen11Z@users.noreply.github.com>
Date: Thu, 13 Jun 2024 23:36:43 +0800
Subject: [PATCH] Fix squeeze bug

---
 wired_table_rec/utils_table_recover.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wired_table_rec/utils_table_recover.py b/wired_table_rec/utils_table_recover.py
index 151f525..10ddc0e 100644
--- 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 是重合度过高的
         # 合并，取两者各个点的最大值