Skip to content
Toggle navigation
Toggle navigation
This project
Loading...
Sign in
周伟奇
/
bmw-ocr
Go to a project
Toggle navigation
Toggle navigation pinning
Projects
Groups
Snippets
Help
Project
Activity
Repository
Graphs
Network
Create a new issue
Commits
Issue Boards
Files
Commits
Network
Compare
Branches
Tags
5eb7c67e
authored
2024-04-28 15:07:22 +0800
by
冯轩
Browse Files
Options
Browse Files
Tag
Download
Email Patches
Plain Diff
ADD:log
1 parent
35069885
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
3 deletions
src/apps/doc/management/commands/ocr_process.py
src/apps/doc/views.py
src/common/electronic_afc_contract/afc_contract_ocr.py
src/apps/doc/management/commands/ocr_process.py
View file @
5eb7c67
...
...
@@ -1494,7 +1494,7 @@ class Command(BaseCommand, LoggerMixin):
# AFC合同
if
classify_1_str
==
str
(
consts
.
CONTRACT_CLASSIFY
):
is_fsm
=
doc
.
is_ovp_fsm
==
1
ocr_result
=
afc_predict
(
pdf_handler
.
pdf_info
,
is_fsm
=
is_fsm
)
ocr_result
=
afc_predict
(
self
,
pdf_handler
.
pdf_info
,
is_fsm
=
is_fsm
)
page_res
=
{}
for
page_num
,
page_info
in
ocr_result
.
get
(
'page_info'
,
{})
.
items
():
if
isinstance
(
page_num
,
str
)
and
page_num
.
startswith
(
'page_'
):
...
...
src/apps/doc/views.py
View file @
5eb7c67
...
...
@@ -1117,7 +1117,7 @@ class DocView(DocGenericView, DocHandler):
data_source
=
request
.
POST
.
get
(
'data_source'
,
''
)
document_name
=
request
.
POST
.
get
(
'document_name'
,
''
)
fsm
=
request
.
POST
.
get
(
'is_fsm'
,
'N'
)
args
=
{
'business_type'
:
business_type
,
'document_scheme'
:
document_scheme
,
'data_source'
:
data_source
,
'document_name'
:
document_name
,
'pdf_file'
:
pdf_file
}
args
=
{
'business_type'
:
business_type
,
'document_scheme'
:
document_scheme
,
'data_source'
:
data_source
,
'document_name'
:
document_name
,
'pdf_file'
:
pdf_file
,
'fsm'
:
fsm
}
# business_type = random.choice(consts.BUSINESS_TYPE_LIST)
# business_type = consts.BUSINESS_TYPE_LIST[0]
tmp_save_path
=
os
.
path
.
join
(
conf
.
DATA_DIR
,
business_type
,
'{0}.pdf'
.
format
(
metadata_version_id
))
...
...
src/common/electronic_afc_contract/afc_contract_ocr.py
View file @
5eb7c67
...
...
@@ -24,7 +24,7 @@ def extract_info(ocr_results):
return
{
'page_1'
:
{
'合同编号'
:
contract_no
}}
def
predict
(
pdf_info
,
is_qrs
=
False
,
is_fsm
=
False
):
def
predict
(
self
,
pdf_info
,
is_qrs
=
False
,
is_fsm
=
False
):
pop_seceond_page_info
=
{}
if
not
is_fsm
and
not
is_qrs
and
len
(
pdf_info
)
==
9
:
pop_seceond_page_info
=
pdf_info
.
pop
(
'1'
,
{})
...
...
@@ -61,8 +61,10 @@ def predict(pdf_info, is_qrs=False, is_fsm=False):
else
:
# 输入是整个 PDF 中的信息
if
is_fsm
:
self
.
online_log
.
info
(
'afc_contract_is_fsm'
)
f
=
FSMFinder
(
pdf_info
,
ocr_results
=
ocr_results
)
else
:
self
.
online_log
.
info
(
'afc_contract_is_non_fsm'
)
f
=
Finder
(
pdf_info
,
ocr_results
=
ocr_results
)
results
=
f
.
get_info
()
return
results
...
...
Write
Preview
Styling with
Markdown
is supported
Attach a file
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 post a comment