magic-pdf 0.10.1__py3-none-any.whl → 0.10.2__py3-none-any.whl

magic_pdf/pre_proc/detect_footer_header_by_statistics.py

@@ -1,284 +0,0 @@
-from collections import defaultdict
-
-from magic_pdf.libs.boxbase import calculate_iou
-
-
-def compare_bbox_with_list(bbox, bbox_list, tolerance=1):
-    return any(all(abs(a - b) < tolerance for a, b in zip(bbox, common_bbox)) for common_bbox in bbox_list)
-
-def is_single_line_block(block):
-    # Determine based on the width and height of the block
-    block_width = block["X1"] - block["X0"]
-    block_height = block["bbox"][3] - block["bbox"][1]
-
-    # If the height of the block is close to the average character height and the width is large, it is considered a single line
-    return block_height <= block["avg_char_height"] * 3 and block_width > block["avg_char_width"] * 3
-
-def get_most_common_bboxes(bboxes, page_height, position="top", threshold=0.25, num_bboxes=3, min_frequency=2):
-    """
-    This function gets the most common bboxes from the bboxes
-
-    Parameters
-    ----------
-    bboxes : list
-        bboxes
-    page_height : float
-        height of the page
-    position : str, optional
-        "top" or "bottom", by default "top"
-    threshold : float, optional
-        threshold, by default 0.25
-    num_bboxes : int, optional
-        number of bboxes to return, by default 3
-    min_frequency : int, optional
-        minimum frequency of the bbox, by default 2
-
-    Returns
-    -------
-    common_bboxes : list
-        common bboxes
-    """
-    # Filter bbox by position
-    if position == "top":
-        filtered_bboxes = [bbox for bbox in bboxes if bbox[1] < page_height * threshold]
-    else:
-        filtered_bboxes = [bbox for bbox in bboxes if bbox[3] > page_height * (1 - threshold)]
-
-    # Find the most common bbox
-    bbox_count = defaultdict(int)
-    for bbox in filtered_bboxes:
-        bbox_count[tuple(bbox)] += 1
-
-    # Get the most frequently occurring bbox, but only consider it when the frequency exceeds min_frequency
-    common_bboxes = [
-        bbox for bbox, count in sorted(bbox_count.items(), key=lambda item: item[1], reverse=True) if count >= min_frequency
-    ][:num_bboxes]
-    return common_bboxes
-
-def detect_footer_header2(result_dict, similarity_threshold=0.5):
-    """
-    This function detects the header and footer of the document.
-
-    Parameters
-    ----------
-    result_dict : dict
-        result dictionary
-
-    Returns
-    -------
-    result_dict : dict
-        result dictionary
-    """
-    # Traverse all blocks in the document
-    single_line_blocks = 0
-    total_blocks = 0
-    single_line_blocks = 0
-
-    for page_id, blocks in result_dict.items():
-        if page_id.startswith("page_"):
-            for block_key, block in blocks.items():
-                if block_key.startswith("block_"):
-                    total_blocks += 1
-                    if is_single_line_block(block):
-                        single_line_blocks += 1
-
-    # If there are no blocks, skip the header and footer detection
-    if total_blocks == 0:
-        print("No blocks found. Skipping header/footer detection.")
-        return result_dict
-
-    # If most of the blocks are single-line, skip the header and footer detection
-    if single_line_blocks / total_blocks > 0.5:  # 50% of the blocks are single-line
-        # print("Skipping header/footer detection for text-dense document.")
-        return result_dict
-
-    # Collect the bounding boxes of all blocks
-    all_bboxes = []
-    all_texts = []
-
-    for page_id, blocks in result_dict.items():
-        if page_id.startswith("page_"):
-            for block_key, block in blocks.items():
-                if block_key.startswith("block_"):
-                    all_bboxes.append(block["bbox"])
-
-    # Get the height of the page
-    page_height = max(bbox[3] for bbox in all_bboxes)
-
-    # Get the most common bbox lists for headers and footers
-    common_header_bboxes = get_most_common_bboxes(all_bboxes, page_height, position="top") if all_bboxes else []
-    common_footer_bboxes = get_most_common_bboxes(all_bboxes, page_height, position="bottom") if all_bboxes else []
-
-    # Detect and mark headers and footers
-    for page_id, blocks in result_dict.items():
-        if page_id.startswith("page_"):
-            for block_key, block in blocks.items():
-                if block_key.startswith("block_"):
-                    bbox = block["bbox"]
-                    text = block["text"]
-
-                    is_header = compare_bbox_with_list(bbox, common_header_bboxes)
-                    is_footer = compare_bbox_with_list(bbox, common_footer_bboxes)
-                    block["is_header"] = int(is_header)
-                    block["is_footer"] = int(is_footer)
-
-    return result_dict
-
-
-def __get_page_size(page_sizes:list):
-    """
-    页面大小可能不一样
-    """
-    w = sum([w for w,h in page_sizes])/len(page_sizes)
-    h = sum([h for w,h  in page_sizes])/len(page_sizes)
-    return w, h
-
-def __calculate_iou(bbox1, bbox2):
-    iou = calculate_iou(bbox1, bbox2)
-    return iou
-
-def __is_same_pos(box1, box2, iou_threshold):
-    iou = __calculate_iou(box1, box2)
-    return iou >= iou_threshold
-
-
-def get_most_common_bbox(bboxes:list, page_size:list, page_cnt:int,  page_range_threshold=0.2, iou_threshold=0.9):
-    """
-    common bbox必须大于page_cnt的1/3
-    """
-    min_occurance_cnt = max(3, page_cnt//4)
-    header_det_bbox = []
-    footer_det_bbox = []
-    
-    hdr_same_pos_group = []
-    btn_same_pos_group = []
-    
-    page_w, page_h = __get_page_size(page_size)
-    top_y, bottom_y = page_w*page_range_threshold, page_h*(1-page_range_threshold)
-    
-    top_bbox = [b for b in bboxes if b[3]<top_y]
-    bottom_bbox = [b for b in bboxes if b[1]>bottom_y]
-    # 然后开始排序，寻找最经常出现的bbox, 寻找的时候如果IOU>iou_threshold就算是一个
-    for i in range(0, len(top_bbox)):
-        hdr_same_pos_group.append([top_bbox[i]])
-        for j in range(i+1, len(top_bbox)):
-            if __is_same_pos(top_bbox[i], top_bbox[j], iou_threshold):
-                #header_det_bbox = [min(top_bbox[i][0], top_bbox[j][0]), min(top_bbox[i][1], top_bbox[j][1]), max(top_bbox[i][2], top_bbox[j][2]), max(top_bbox[i][3],top_bbox[j][3])]
-                hdr_same_pos_group[i].append(top_bbox[j])
-                
-    for i in range(0, len(bottom_bbox)):
-        btn_same_pos_group.append([bottom_bbox[i]])
-        for j in range(i+1, len(bottom_bbox)):
-            if __is_same_pos(bottom_bbox[i], bottom_bbox[j], iou_threshold):
-                #footer_det_bbox = [min(bottom_bbox[i][0], bottom_bbox[j][0]), min(bottom_bbox[i][1], bottom_bbox[j][1]), max(bottom_bbox[i][2], bottom_bbox[j][2]), max(bottom_bbox[i][3],bottom_bbox[j][3])]
-                btn_same_pos_group[i].append(bottom_bbox[j])
-                
-    # 然后看下每一组的bbox，是否符合大于page_cnt一定比例
-    hdr_same_pos_group = [g for g in hdr_same_pos_group if len(g)>=min_occurance_cnt]
-    btn_same_pos_group = [g for g in btn_same_pos_group if len(g)>=min_occurance_cnt]
-    
-    # 平铺2个list[list]
-    hdr_same_pos_group = [bbox for g in hdr_same_pos_group for bbox in g]
-    btn_same_pos_group = [bbox for g in btn_same_pos_group for bbox in g]
-    # 寻找hdr_same_pos_group中的box[3]最大值，btn_same_pos_group中的box[1]最小值
-    hdr_same_pos_group.sort(key=lambda b:b[3])
-    btn_same_pos_group.sort(key=lambda b:b[1])
-    
-    hdr_y = hdr_same_pos_group[-1][3] if hdr_same_pos_group else 0
-    btn_y = btn_same_pos_group[0][1] if btn_same_pos_group else page_h
-    
-    header_det_bbox = [0, 0, page_w, hdr_y]
-    footer_det_bbox = [0, btn_y, page_w, page_h]
-    # logger.warning(f"header: {header_det_bbox}, footer: {footer_det_bbox}")
-    return header_det_bbox, footer_det_bbox, page_w, page_h
-    
-
-def drop_footer_header(pdf_info_dict:dict):
-    """
-    启用规则探测,在全局的视角上通过统计的方法。
-    """
-    header = []
-    footer = []
-    
-    all_text_bboxes = [blk['bbox'] for _, val in pdf_info_dict.items() for blk in val['preproc_blocks']]
-    image_bboxes = [img['bbox'] for _, val in pdf_info_dict.items() for img in val['images']] + [img['bbox'] for _, val in pdf_info_dict.items() for img in val['image_backup']]
-    page_size = [val['page_size'] for _, val in pdf_info_dict.items()]
-    page_cnt = len(pdf_info_dict.keys()) # 一共多少页
-    header, footer, page_w, page_h = get_most_common_bbox(all_text_bboxes+image_bboxes, page_size, page_cnt)
-    
-    """"
-    把范围扩展到页面水平的整个方向上
-    """        
-    if header:
-        header = [0, 0, page_w, header[3]+1]
-        
-    if footer:
-        footer = [0, footer[1]-1, page_w, page_h]
-        
-    # 找到footer, header范围之后，针对每一页pdf，从text、图片中删除这些范围内的内容
-    # 移除text block
-    
-    for _, page_info in pdf_info_dict.items():
-        header_text_blk = []
-        footer_text_blk = []
-        for blk in page_info['preproc_blocks']:
-            blk_bbox = blk['bbox']
-            if header and blk_bbox[3]<=header[3]:
-                blk['tag'] = "header"
-                header_text_blk.append(blk)
-            elif footer and blk_bbox[1]>=footer[1]:
-                blk['tag'] = "footer"
-                footer_text_blk.append(blk)
-                
-        # 放入text_block_droped中
-        page_info['droped_text_block'].extend(header_text_blk)
-        page_info['droped_text_block'].extend(footer_text_blk)
-        
-        for blk in header_text_blk:
-            page_info['preproc_blocks'].remove(blk)
-        for blk in footer_text_blk:
-            page_info['preproc_blocks'].remove(blk)
-            
-        """接下来把footer、header上的图片也删除掉。图片包括正常的和backup的"""
-        header_image = []
-        footer_image = []
-        
-        for image_info in page_info['images']:
-            img_bbox = image_info['bbox']
-            if header and img_bbox[3]<=header[3]:
-                image_info['tag'] = "header"
-                header_image.append(image_info)
-            elif footer and img_bbox[1]>=footer[1]:
-                image_info['tag'] = "footer"
-                footer_image.append(image_info)
-                
-        page_info['droped_image_block'].extend(header_image)
-        page_info['droped_image_block'].extend(footer_image)
-        
-        for img in header_image:
-            page_info['images'].remove(img)
-        for img in footer_image:
-            page_info['images'].remove(img)
-            
-        """接下来吧backup的图片也删除掉"""
-        header_image = []
-        footer_image = []
-        
-        for image_info in page_info['image_backup']:
-            img_bbox = image_info['bbox']
-            if header and img_bbox[3]<=header[3]:
-                image_info['tag'] = "header"
-                header_image.append(image_info)
-            elif footer and img_bbox[1]>=footer[1]:
-                image_info['tag'] = "footer"
-                footer_image.append(image_info)
-                
-        page_info['droped_image_block'].extend(header_image)
-        page_info['droped_image_block'].extend(footer_image)
-        
-        for img in header_image:
-            page_info['image_backup'].remove(img)
-        for img in footer_image:
-            page_info['image_backup'].remove(img)
-            
-    return header, footer

magic_pdf/pre_proc/detect_footnote.py

@@ -1,170 +0,0 @@
-from collections import Counter
-from magic_pdf.libs.commons import fitz             # pyMuPDF库
-from magic_pdf.libs.coordinate_transform import get_scale_ratio
-
-
-def parse_footnotes_by_model(page_ID: int, page: fitz.Page, json_from_DocXchain_obj: dict, md_bookname_save_path=None, debug_mode=False):
-    """
-    :param page_ID: int类型，当前page在当前pdf文档中是第page_D页。
-    :param page :fitz读取的当前页的内容
-    :param res_dir_path: str类型，是每一个pdf文档，在当前.py文件的目录下生成一个与pdf文档同名的文件夹，res_dir_path就是文件夹的dir
-    :param json_from_DocXchain_obj: dict类型，把pdf文档送入DocXChain模型中后，提取bbox，结果保存到pdf文档同名文件夹下的 page_ID.json文件中了。json_from_DocXchain_obj就是打开后的dict
-    """
-
-    #--------- 通过json_from_DocXchain来获取 footnote ---------#
-    footnote_bbox_from_DocXChain = []
-
-    xf_json = json_from_DocXchain_obj
-    horizontal_scale_ratio, vertical_scale_ratio = get_scale_ratio(xf_json, page)
-
-    # {0: 'title',  # 标题
-    # 1: 'figure', # 图片
-    #  2: 'plain text',  # 文本
-    #  3: 'header',      # 页眉
-    #  4: 'page number', # 页码
-    #  5: 'footnote',    # 脚注
-    #  6: 'footer',      # 页脚
-    #  7: 'table',       # 表格
-    #  8: 'table caption',  # 表格描述
-    #  9: 'figure caption', # 图片描述
-    #  10: 'equation',      # 公式
-    #  11: 'full column',   # 单栏
-    #  12: 'sub column',    # 多栏
-    #  13: 'embedding',     # 嵌入公式
-    #  14: 'isolated'}      # 单行公式
-    for xf in xf_json['layout_dets']:
-        L = xf['poly'][0] / horizontal_scale_ratio
-        U = xf['poly'][1] / vertical_scale_ratio
-        R = xf['poly'][2] / horizontal_scale_ratio
-        D = xf['poly'][5] / vertical_scale_ratio
-        # L += pageL          # 有的页面，artBox偏移了。不在（0,0）
-        # R += pageL
-        # U += pageU
-        # D += pageU
-        L, R = min(L, R), max(L, R)
-        U, D = min(U, D), max(U, D)
-        # if xf['category_id'] == 5 and xf['score'] >= 0.3:
-        if xf['category_id'] == 5 and xf['score'] >= 0.43:  # 新的footnote阈值
-            footnote_bbox_from_DocXChain.append((L, U, R, D))
-            
-    
-    footnote_final_names = []
-    footnote_final_bboxs = []
-    footnote_ID = 0
-    for L, U, R, D in footnote_bbox_from_DocXChain:
-        if debug_mode:
-            # cur_footnote = page.get_pixmap(clip=(L,U,R,D))
-            new_footnote_name = "footnote_{}_{}.png".format(page_ID, footnote_ID)    # 脚注name
-            # cur_footnote.save(md_bookname_save_path + '/' + new_footnote_name)           # 把脚注存储在新建的文件夹，并命名
-            footnote_final_names.append(new_footnote_name)                        # 把脚注的名字存在list中
-        footnote_final_bboxs.append((L, U, R, D))
-        footnote_ID += 1
-        
-
-    footnote_final_bboxs.sort(key = lambda LURD: (LURD[1], LURD[0]))
-    curPage_all_footnote_bboxs = footnote_final_bboxs
-    return curPage_all_footnote_bboxs
-
-
-def need_remove(block):
-    if 'lines' in block and len(block['lines']) > 0:
-        # block中只有一行，且该行文本全是大写字母，或字体为粗体bold关键词，SB关键词，把这个block捞回来
-        if len(block['lines']) == 1:
-            if 'spans' in block['lines'][0] and len(block['lines'][0]['spans']) == 1:
-                font_keywords = ['SB', 'bold', 'Bold']
-                if block['lines'][0]['spans'][0]['text'].isupper() or any(keyword in block['lines'][0]['spans'][0]['font'] for keyword in font_keywords):
-                    return True
-        for line in block['lines']:
-            if 'spans' in line and len(line['spans']) > 0:
-                for span in line['spans']:
-                    # 检测"keyword"是否在span中，忽略大小写
-                    if "keyword" in span['text'].lower():
-                        return True
-    return False
-
-def parse_footnotes_by_rule(remain_text_blocks, page_height, page_id, main_text_font):
-    """
-    根据给定的文本块、页高和页码，解析出符合规则的脚注文本块，并返回其边界框。
-
-    Args:
-        remain_text_blocks (list): 包含所有待处理的文本块的列表。
-        page_height (float): 页面的高度。
-        page_id (int): 页面的ID。
-
-    Returns:
-        list: 符合规则的脚注文本块的边界框列表。
-
-    """
-    # if page_id > 20:
-    if page_id > 2:  # 为保证精确度，先只筛选前3页
-        return []
-    else:
-        # 存储每一行的文本块大小的列表
-        line_sizes = []
-        # 存储每个文本块的平均行大小
-        block_sizes = []
-        # 存储每一行的字体信息
-        # font_names = []
-        font_names = Counter()
-        if len(remain_text_blocks) > 0:
-            for block in remain_text_blocks:
-                block_line_sizes = []
-                # block_fonts = []
-                block_fonts = Counter()
-                for line in block['lines']:
-                    # 提取每个span的size属性，并计算行大小
-                    span_sizes = [span['size'] for span in line['spans'] if 'size' in span]
-                    if span_sizes:
-                        line_size = sum(span_sizes) / len(span_sizes)
-                        line_sizes.append(line_size)
-                        block_line_sizes.append(line_size)
-                    span_font = [(span['font'], len(span['text'])) for span in line['spans'] if 'font' in span and len(span['text']) > 0]
-                    if span_font:
-                        #  main_text_font应该用基于字数最多的字体而不是span级别的统计
-                        # font_names.append(font_name for font_name in span_font)
-                        # block_fonts.append(font_name for font_name in span_font)
-                        for font, count in span_font:
-                            # font_names.extend([font] * count)
-                            # block_fonts.extend([font] * count)
-                            font_names[font] += count
-                            block_fonts[font] += count
-                if block_line_sizes:
-                    # 计算文本块的平均行大小
-                    block_size = sum(block_line_sizes) / len(block_line_sizes)
-                    # block_font = collections.Counter(block_fonts).most_common(1)[0][0]
-                    block_font = block_fonts.most_common(1)[0][0]
-                    block_sizes.append((block, block_size, block_font))
-
-            # 计算main_text_size
-            main_text_size = Counter(line_sizes).most_common(1)[0][0]
-            # 计算main_text_font
-            # main_text_font = collections.Counter(font_names).most_common(1)[0][0]
-            # main_text_font = font_names.most_common(1)[0][0]
-            # 删除一些可能被误识别为脚注的文本块
-            block_sizes = [(block, block_size, block_font) for block, block_size, block_font in block_sizes if not need_remove(block)]
-
-            # 检测footnote_block 并返回 footnote_bboxes
-            # footnote_bboxes = [block['bbox'] for block, block_size, block_font in block_sizes if
-            #                    block['bbox'][1] > page_height * 0.6 and block_size < main_text_size
-            #                    and (len(block['lines']) < 5 or block_font != main_text_font)]
-                               # and len(block['lines']) < 5]
-            footnote_bboxes = [block['bbox'] for block, block_size, block_font in block_sizes if
-                               block['bbox'][1] > page_height * 0.6 and
-                               #  较为严格的规则
-                               block_size < main_text_size and
-                               (len(block['lines']) < 5 or
-                                block_font != main_text_font)]
-
-                               #  较为宽松的规则
-                               # sum([block_size < main_text_size,
-                               #      len(block['lines']) < 5,
-                               #      block_font != main_text_font])
-                               # >= 2]
-
-
-            return footnote_bboxes
-        else:
-            return []
-
-
-

magic_pdf/pre_proc/detect_header.py

@@ -1,64 +0,0 @@
-from magic_pdf.libs.commons import fitz             # pyMuPDF库
-from magic_pdf.libs.coordinate_transform import get_scale_ratio
-
-
-def parse_headers(page_ID: int, page: fitz.Page, json_from_DocXchain_obj: dict):
-    """
-    :param page_ID: int类型，当前page在当前pdf文档中是第page_D页。
-    :param page :fitz读取的当前页的内容
-    :param res_dir_path: str类型，是每一个pdf文档，在当前.py文件的目录下生成一个与pdf文档同名的文件夹，res_dir_path就是文件夹的dir
-    :param json_from_DocXchain_obj: dict类型，把pdf文档送入DocXChain模型中后，提取bbox，结果保存到pdf文档同名文件夹下的 page_ID.json文件中了。json_from_DocXchain_obj就是打开后的dict
-    """
-
-    #--------- 通过json_from_DocXchain来获取 header ---------#
-    header_bbox_from_DocXChain = []
-
-    xf_json = json_from_DocXchain_obj
-    horizontal_scale_ratio, vertical_scale_ratio = get_scale_ratio(xf_json, page)
-
-    # {0: 'title',  # 标题
-    # 1: 'figure', # 图片
-    #  2: 'plain text',  # 文本
-    #  3: 'header',      # 页眉
-    #  4: 'page number', # 页码
-    #  5: 'footnote',    # 脚注
-    #  6: 'footer',      # 页脚
-    #  7: 'table',       # 表格
-    #  8: 'table caption',  # 表格描述
-    #  9: 'figure caption', # 图片描述
-    #  10: 'equation',      # 公式
-    #  11: 'full column',   # 单栏
-    #  12: 'sub column',    # 多栏
-    #  13: 'embedding',     # 嵌入公式
-    #  14: 'isolated'}      # 单行公式
-    for xf in xf_json['layout_dets']:
-        L = xf['poly'][0] / horizontal_scale_ratio
-        U = xf['poly'][1] / vertical_scale_ratio
-        R = xf['poly'][2] / horizontal_scale_ratio
-        D = xf['poly'][5] / vertical_scale_ratio
-        # L += pageL          # 有的页面，artBox偏移了。不在（0,0）
-        # R += pageL
-        # U += pageU
-        # D += pageU
-        L, R = min(L, R), max(L, R)
-        U, D = min(U, D), max(U, D)
-        if xf['category_id'] == 3 and xf['score'] >= 0.3:
-            header_bbox_from_DocXChain.append((L, U, R, D))
-            
-    
-    header_final_names = []
-    header_final_bboxs = []
-    header_ID = 0
-    for L, U, R, D in header_bbox_from_DocXChain:
-        # cur_header = page.get_pixmap(clip=(L,U,R,D))
-        new_header_name = "header_{}_{}.png".format(page_ID, header_ID)    # 页眉name
-        # cur_header.save(res_dir_path + '/' + new_header_name)           # 把页眉存储在新建的文件夹，并命名
-        header_final_names.append(new_header_name)                        # 把页面的名字存在list中
-        header_final_bboxs.append((L, U, R, D))
-        header_ID += 1
-        
-
-    header_final_bboxs.sort(key = lambda LURD: (LURD[1], LURD[0]))
-    curPage_all_header_bboxs = header_final_bboxs
-    return curPage_all_header_bboxs
-