Skip to content

P
pdf-miner
Commit 8db33db4 authored by 赵小蒙's avatar 赵小蒙
Browse files
Options

add annotation

parent 9a377463
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
magic_pdf/model/magic_model.py
View file @ 8db33db4
...@@ -90,8 +90,11 @@ class MagicModel: ...@@ -90,8 +90,11 @@ class MagicModel:
def __init__(self, model_list: list, docs: fitz.Document): def __init__(self, model_list: list, docs: fitz.Document):
self.__model_list = model_list self.__model_list = model_list
self.__docs = docs self.__docs = docs
'''为所有模型数据添加bbox信息(缩放，poly->bbox)'''
self.__fix_axis() self.__fix_axis()
'''删除置信度特别低的模型数据(<0.05),提高质量'''
self.__fix_by_remove_low_confidence() self.__fix_by_remove_low_confidence()
'''删除高iou(>0.9)数据中置信度较低的那个'''
self.__fix_by_remove_high_iou_and_low_confidence() self.__fix_by_remove_high_iou_and_low_confidence()
def __reduct_overlap(self, bboxes): def __reduct_overlap(self, bboxes):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment