construct_page_dict.py 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. def construct_page_component(page_id, image_info, table_info, text_blocks_preproc, layout_bboxes, inline_eq_info, interline_eq_info, raw_pymu_blocks,
  2. removed_text_blocks, removed_image_blocks, images_backup, droped_table_block, table_backup,layout_tree,
  3. page_w, page_h, footnote_bboxes_tmp):
  4. """
  5. """
  6. return_dict = {}
  7. return_dict['para_blocks'] = {}
  8. return_dict['preproc_blocks'] = text_blocks_preproc
  9. return_dict['images'] = image_info
  10. return_dict['tables'] = table_info
  11. return_dict['interline_equations'] = interline_eq_info
  12. return_dict['inline_equations'] = inline_eq_info
  13. return_dict['layout_bboxes'] = layout_bboxes
  14. return_dict['pymu_raw_blocks'] = raw_pymu_blocks
  15. return_dict['global_statistic'] = {}
  16. return_dict['droped_text_block'] = removed_text_blocks
  17. return_dict['droped_image_block'] = removed_image_blocks
  18. return_dict['droped_table_block'] = []
  19. return_dict['image_backup'] = images_backup
  20. return_dict['table_backup'] = []
  21. return_dict['page_idx'] = page_id
  22. return_dict['page_size'] = [page_w, page_h]
  23. return_dict['_layout_tree'] = layout_tree # 辅助分析layout作用
  24. return_dict['footnote_bboxes_tmp'] = footnote_bboxes_tmp
  25. return return_dict
  26. def ocr_construct_page_component(blocks, layout_bboxes, page_id, page_w, page_h, layout_tree,
  27. images, tables, interline_equations, inline_equations,
  28. dropped_text_block, dropped_image_block, dropped_table_block, dropped_equation_block,
  29. need_remove_spans_bboxes_dict):
  30. return_dict = {
  31. 'preproc_blocks': blocks,
  32. 'layout_bboxes': layout_bboxes,
  33. 'page_idx': page_id,
  34. 'page_size': [page_w, page_h],
  35. '_layout_tree': layout_tree,
  36. 'images': images,
  37. 'tables': tables,
  38. 'interline_equations': interline_equations,
  39. 'inline_equations': inline_equations,
  40. 'droped_text_block': dropped_text_block,
  41. 'droped_image_block': dropped_image_block,
  42. 'droped_table_block': dropped_table_block,
  43. 'dropped_equation_block': dropped_equation_block,
  44. 'droped_bboxes': need_remove_spans_bboxes_dict,
  45. }
  46. return return_dict