Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
weixin_41840029
PaddleOCR
提交
b26ce237
P
PaddleOCR
项目概览
weixin_41840029
/
PaddleOCR
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleOCR
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
b26ce237
编写于
8月 16, 2022
作者:
文幕地方
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
rm unused code
上级
bb53c8d1
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
9 addition
and
18 deletion
+9
-18
paddleocr.py
paddleocr.py
+9
-18
未找到文件。
paddleocr.py
浏览文件 @
b26ce237
...
@@ -318,10 +318,10 @@ def parse_args(mMain=True):
...
@@ -318,10 +318,10 @@ def parse_args(mMain=True):
"--structure_version"
,
"--structure_version"
,
type
=
str
,
type
=
str
,
choices
=
SUPPORT_STRUCTURE_MODEL_VERSION
,
choices
=
SUPPORT_STRUCTURE_MODEL_VERSION
,
default
=
'PP-Structure'
,
default
=
'PP-Structure
v2
'
,
help
=
'Model version, the current model support list is as follows:'
help
=
'Model version, the current model support list is as follows:'
' 1. PP-Structure Support en table structure model.'
' 1. PP-Structure Support en table structure model.'
' 2. PP-Structure Support ch and en table structure model.'
)
' 2. PP-Structure
v2
Support ch and en table structure model.'
)
for
action
in
parser
.
_actions
:
for
action
in
parser
.
_actions
:
if
action
.
dest
in
[
if
action
.
dest
in
[
...
@@ -529,6 +529,12 @@ class PPStructure(StructureSystem):
...
@@ -529,6 +529,12 @@ class PPStructure(StructureSystem):
if
not
params
.
show_log
:
if
not
params
.
show_log
:
logger
.
setLevel
(
logging
.
INFO
)
logger
.
setLevel
(
logging
.
INFO
)
lang
,
det_lang
=
parse_lang
(
params
.
lang
)
lang
,
det_lang
=
parse_lang
(
params
.
lang
)
if
lang
==
'ch'
:
table_lang
=
'ch'
else
:
table_lang
=
'en'
if
params
.
structure_version
==
'PP-Structure'
:
params
.
merge_no_span_structure
=
False
# init model dir
# init model dir
det_model_config
=
get_model_config
(
'OCR'
,
params
.
ocr_version
,
'det'
,
det_model_config
=
get_model_config
(
'OCR'
,
params
.
ocr_version
,
'det'
,
...
@@ -543,7 +549,7 @@ class PPStructure(StructureSystem):
...
@@ -543,7 +549,7 @@ class PPStructure(StructureSystem):
params
.
rec_model_dir
,
params
.
rec_model_dir
,
os
.
path
.
join
(
BASE_DIR
,
'whl'
,
'rec'
,
lang
),
rec_model_config
[
'url'
])
os
.
path
.
join
(
BASE_DIR
,
'whl'
,
'rec'
,
lang
),
rec_model_config
[
'url'
])
table_model_config
=
get_model_config
(
table_model_config
=
get_model_config
(
'STRUCTURE'
,
params
.
structure_version
,
'table'
,
'ch'
)
'STRUCTURE'
,
params
.
structure_version
,
'table'
,
table_lang
)
params
.
table_model_dir
,
table_url
=
confirm_model_dir_url
(
params
.
table_model_dir
,
table_url
=
confirm_model_dir_url
(
params
.
table_model_dir
,
params
.
table_model_dir
,
os
.
path
.
join
(
BASE_DIR
,
'whl'
,
'table'
),
table_model_config
[
'url'
])
os
.
path
.
join
(
BASE_DIR
,
'whl'
,
'table'
),
table_model_config
[
'url'
])
...
@@ -629,18 +635,3 @@ def main():
...
@@ -629,18 +635,3 @@ def main():
for
item
in
result
:
for
item
in
result
:
item
.
pop
(
'img'
)
item
.
pop
(
'img'
)
logger
.
info
(
item
)
logger
.
info
(
item
)
if
__name__
==
"__main__"
:
table_engine
=
PPStructure
(
layout
=
False
,
show_log
=
True
)
save_folder
=
'./output'
img_path
=
'ppstructure/docs/table/table.jpg'
img
=
cv2
.
imread
(
img_path
)
result
=
table_engine
(
img
)
save_structure_res
(
result
,
save_folder
,
os
.
path
.
basename
(
img_path
).
split
(
'.'
)[
0
])
for
line
in
result
:
line
.
pop
(
'img'
)
print
(
line
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录