Minor changes

app.py

@@ -211,7 +211,7 @@ def convert_stucture(page_tokens, pil_img, structure_result):
     # print('table_class_objects:', table_class_objects)
     # print('table_bbox:', table_bbox)
 
-    tokens_in_table = [token for token in page_tokens if postprocess.iob(token['bbox'], table_bbox) >= 0.5]
+    tokens_in_table = [token for token in page_tokens if postprocess.iob(token['bbox'], table_bbox) >= 0.01]
     # print('tokens_in_table:', tokens_in_table)
 
     table_structures, cells, confidence_score = postprocess.objects_to_cells(table, table_objects, tokens_in_table, structure_class_names, structure_class_thresholds)

postprocess.py

@@ -199,8 +199,8 @@ def nms_by_containment(container_objects, package_objects, overlap_threshold=0.5
             if not suppression[object1_num]:
                 object1_packages = set(packages_by_container[object1_num])
                 if len(object2_packages.intersection(object1_packages)) > 0 \
-                    and (iob(container_objects[object2_num]['bbox'], container_objects[object1_num]['bbox']) > 0.5 \
-                         or iob(container_objects[object1_num]['bbox'], container_objects[object2_num]['bbox']) > 0.5):
+                    and (iob(container_objects[object2_num]['bbox'], container_objects[object1_num]['bbox']) >= 0.5 \
+                         or iob(container_objects[object1_num]['bbox'], container_objects[object2_num]['bbox']) >= 0.5):
                     suppression[object2_num] = True
 
     final_objects = [obj for idx, obj in enumerate(container_objects) if not suppression[idx]]