Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
P
pdf-miner
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Qin Kaijie
pdf-miner
Commits
46c5d6e0
Unverified
Commit
46c5d6e0
authored
Apr 23, 2024
by
myhloli
Committed by
GitHub
Apr 23, 2024
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #58 from myhloli/master
模型数据初始化时根据置信度进行过滤,预设阈值95%
parents
23bacc60
c460be91
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
3 deletions
+2
-3
magic_model.py
magic_pdf/model/magic_model.py
+2
-3
No files found.
magic_pdf/model/magic_model.py
View file @
46c5d6e0
...
...
@@ -48,7 +48,7 @@ class MagicModel:
need_remove_list
=
[]
layout_dets
=
model_page_info
[
"layout_dets"
]
for
layout_det
in
layout_dets
:
if
layout_det
[
"score"
]
<
0.
6
:
if
layout_det
[
"score"
]
<
0.
95
:
need_remove_list
.
append
(
layout_det
)
else
:
continue
...
...
@@ -59,8 +59,7 @@ class MagicModel:
self
.
__model_list
=
model_list
self
.
__docs
=
docs
self
.
__fix_axis
()
#@TODO 删除掉一些低置信度的会导致分段错误,后面再修复
# self.__fix_by_confidence()
self
.
__fix_by_confidence
()
def
__reduct_overlap
(
self
,
bboxes
):
N
=
len
(
bboxes
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment