fix bug

周伟奇
Showing 1 changed file with 2 additions and 2 deletions
general_extractor/retriever.py
--- a/general_extractor/retriever.py
View file @1942a9b
+++ b/general_extractor/retriever.py
View file @1942a9b
@@ -152,7 +152,7 @@ class Retriever:
            self.rebuild_xy(*key_scope_tuple, cos, sin)[0]
        all_find_value_list = []
-        for go_key_idx, ((x0, y0, x1, y1, x2, y3, x3, y3), text) in go_res.items():
+        for go_key_idx, ((x0, y0, x1, y1, x2, y2, x3, y3), text) in go_res.items():
            cent_x, cent_y = self.rebuild_xy(x0 + ((x2 - x0) / 2), y0 + ((y2 - y0) / 2), cos, sin)
            # if go_key_idx == '98' and key_idx == 34:
            #     print(cent_x)
@@ -226,7 +226,7 @@ class Retriever:
            *key_scope_tuple, cos, sin)[0]
        all_find_value_list = []
-        for go_key_idx, ((x0, y0, x1, y1, x2, y3, x3, y3), text) in go_res.items():
+        for go_key_idx, ((x0, y0, x1, y1, x2, y2, x3, y3), text) in go_res.items():
            cent_x, cent_y = self.rebuild_xy(x0 + ((x2 - x0) / 2), y0 + ((y2 - y0) / 2), cos, sin)
            # if go_key_idx == '98' and key_idx == 34:
            #     print(cent_x)