| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- def construct_page_component(page_id, image_info, table_info, text_blocks_preproc, layout_bboxes, inline_eq_info,
- interline_eq_info, raw_pymu_blocks,
- removed_text_blocks, removed_image_blocks, images_backup, droped_table_block, table_backup,
- layout_tree,
- page_w, page_h, footnote_bboxes_tmp):
- """
-
- """
- return_dict = {}
- return_dict['para_blocks'] = {}
- return_dict['preproc_blocks'] = text_blocks_preproc
- return_dict['images'] = image_info
- return_dict['tables'] = table_info
- return_dict['interline_equations'] = interline_eq_info
- return_dict['inline_equations'] = inline_eq_info
- return_dict['layout_bboxes'] = layout_bboxes
- return_dict['pymu_raw_blocks'] = raw_pymu_blocks
- return_dict['global_statistic'] = {}
- return_dict['droped_text_block'] = removed_text_blocks
- return_dict['droped_image_block'] = removed_image_blocks
- return_dict['droped_table_block'] = []
- return_dict['image_backup'] = images_backup
- return_dict['table_backup'] = []
- return_dict['page_idx'] = page_id
- return_dict['page_size'] = [page_w, page_h]
- return_dict['_layout_tree'] = layout_tree # 辅助分析layout作用
- return_dict['footnote_bboxes_tmp'] = footnote_bboxes_tmp
- return return_dict
- def ocr_construct_page_component(blocks, layout_bboxes, page_id, page_w, page_h, layout_tree,
- images, tables, interline_equations, inline_equations,
- dropped_text_block, dropped_image_block, dropped_table_block, dropped_equation_block,
- need_remove_spans_bboxes_dict):
- return_dict = {
- 'preproc_blocks': blocks,
- 'layout_bboxes': layout_bboxes,
- 'page_idx': page_id,
- 'page_size': [page_w, page_h],
- '_layout_tree': layout_tree,
- 'images': images,
- 'tables': tables,
- 'interline_equations': interline_equations,
- 'inline_equations': inline_equations,
- 'droped_text_block': dropped_text_block,
- 'droped_image_block': dropped_image_block,
- 'droped_table_block': dropped_table_block,
- 'dropped_equation_block': dropped_equation_block,
- 'droped_bboxes': need_remove_spans_bboxes_dict,
- }
- return return_dict
- def ocr_construct_page_component_v2(blocks, layout_bboxes, page_id, page_w, page_h, layout_tree,
- images, tables, interline_equations, discarded_blocks, need_drop, drop_reason):
- return_dict = {
- 'preproc_blocks': blocks,
- 'layout_bboxes': layout_bboxes,
- 'page_idx': page_id,
- 'page_size': [page_w, page_h],
- '_layout_tree': layout_tree,
- 'images': images,
- 'tables': tables,
- 'interline_equations': interline_equations,
- 'discarded_blocks': discarded_blocks,
- 'need_drop': need_drop,
- 'drop_reason': drop_reason,
- }
- return return_dict
|
