fix: Fix bug in boundary cells

shahrukhqasim · Nov 20, 2017 · 975caeb · 975caeb
1 parent 49abb5a
commit 975caeb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/table_detect/prepare_dataset.py b/python/table_detect/prepare_dataset.py
@@ -149,8 +149,8 @@ def dump_doc(self, all_tokens, all_tokens_rects, spatial_features, file_name):
                 token_rect['x'] + token_rect['width'] / 2)]
 
             left_rect = all_tokens_rects[int(neighbor_graph[i, 0])]
-            right_rect = all_tokens_rects[int(neighbor_graph[i, 1])]
-            top_rect = all_tokens_rects[int(neighbor_graph[i, 2])]
+            top_rect = all_tokens_rects[int(neighbor_graph[i, 1])]
+            right_rect = all_tokens_rects[int(neighbor_graph[i, 2])]
             bottom_rect = all_tokens_rects[int(neighbor_graph[i, 3])]
 
             if index == 0: