Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
f85b3ea8
Commit
f85b3ea8
authored
Mar 18, 2022
by
WenmuZhou
Browse files
fix bug in whl import fce
parent
496887f8
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
9 additions
and
14 deletions
+9
-14
MANIFEST.in
MANIFEST.in
+1
-1
paddleocr.py
paddleocr.py
+8
-12
tools/infer/utility.py
tools/infer/utility.py
+0
-1
No files found.
MANIFEST.in
View file @
f85b3ea8
include LICENSE
include LICENSE
include README.md
include README.md
recursive-include ppocr/utils *.
txt utility.py logging.py network.py
recursive-include ppocr/utils *.
*
recursive-include ppocr/data *.py
recursive-include ppocr/data *.py
recursive-include ppocr/postprocess *.py
recursive-include ppocr/postprocess *.py
recursive-include tools/infer *.py
recursive-include tools/infer *.py
...
...
paddleocr.py
View file @
f85b3ea8
...
@@ -42,7 +42,7 @@ __all__ = [
...
@@ -42,7 +42,7 @@ __all__ = [
]
]
SUPPORT_DET_MODEL
=
[
'DB'
]
SUPPORT_DET_MODEL
=
[
'DB'
]
VERSION
=
'2.4.0.
1
'
VERSION
=
'2.4.0.
2
'
SUPPORT_REC_MODEL
=
[
'CRNN'
]
SUPPORT_REC_MODEL
=
[
'CRNN'
]
BASE_DIR
=
os
.
path
.
expanduser
(
"~/.paddleocr/"
)
BASE_DIR
=
os
.
path
.
expanduser
(
"~/.paddleocr/"
)
...
@@ -308,20 +308,18 @@ class PaddleOCR(predict_system.TextSystem):
...
@@ -308,20 +308,18 @@ class PaddleOCR(predict_system.TextSystem):
det_lang
)
det_lang
)
params
.
det_model_dir
,
det_url
=
confirm_model_dir_url
(
params
.
det_model_dir
,
det_url
=
confirm_model_dir_url
(
params
.
det_model_dir
,
params
.
det_model_dir
,
os
.
path
.
join
(
BASE_DIR
,
VERSION
,
'ocr
'
,
'det'
,
det_lang
),
os
.
path
.
join
(
BASE_DIR
,
'whl
'
,
'det'
,
det_lang
),
det_model_config
[
'url'
])
det_model_config
[
'url'
])
rec_model_config
=
get_model_config
(
'OCR'
,
params
.
ocr_version
,
'rec'
,
rec_model_config
=
get_model_config
(
'OCR'
,
params
.
ocr_version
,
'rec'
,
lang
)
lang
)
params
.
rec_model_dir
,
rec_url
=
confirm_model_dir_url
(
params
.
rec_model_dir
,
rec_url
=
confirm_model_dir_url
(
params
.
rec_model_dir
,
params
.
rec_model_dir
,
os
.
path
.
join
(
BASE_DIR
,
VERSION
,
'ocr'
,
'rec'
,
lang
),
os
.
path
.
join
(
BASE_DIR
,
'whl'
,
'rec'
,
lang
),
rec_model_config
[
'url'
])
rec_model_config
[
'url'
])
cls_model_config
=
get_model_config
(
'OCR'
,
params
.
ocr_version
,
'cls'
,
cls_model_config
=
get_model_config
(
'OCR'
,
params
.
ocr_version
,
'cls'
,
'ch'
)
'ch'
)
params
.
cls_model_dir
,
cls_url
=
confirm_model_dir_url
(
params
.
cls_model_dir
,
cls_url
=
confirm_model_dir_url
(
params
.
cls_model_dir
,
params
.
cls_model_dir
,
os
.
path
.
join
(
BASE_DIR
,
VERSION
,
'ocr'
,
'cls'
),
os
.
path
.
join
(
BASE_DIR
,
'whl'
,
'cls'
),
cls_model_config
[
'url'
])
cls_model_config
[
'url'
])
# download model
# download model
maybe_download
(
params
.
det_model_dir
,
det_url
)
maybe_download
(
params
.
det_model_dir
,
det_url
)
maybe_download
(
params
.
rec_model_dir
,
rec_url
)
maybe_download
(
params
.
rec_model_dir
,
rec_url
)
...
@@ -412,20 +410,18 @@ class PPStructure(OCRSystem):
...
@@ -412,20 +410,18 @@ class PPStructure(OCRSystem):
det_lang
)
det_lang
)
params
.
det_model_dir
,
det_url
=
confirm_model_dir_url
(
params
.
det_model_dir
,
det_url
=
confirm_model_dir_url
(
params
.
det_model_dir
,
params
.
det_model_dir
,
os
.
path
.
join
(
BASE_DIR
,
VERSION
,
'ocr
'
,
'det'
,
det_lang
),
os
.
path
.
join
(
BASE_DIR
,
'whl
'
,
'det'
,
det_lang
),
det_model_config
[
'url'
])
det_model_config
[
'url'
])
rec_model_config
=
get_model_config
(
'OCR'
,
params
.
ocr_version
,
'rec'
,
rec_model_config
=
get_model_config
(
'OCR'
,
params
.
ocr_version
,
'rec'
,
lang
)
lang
)
params
.
rec_model_dir
,
rec_url
=
confirm_model_dir_url
(
params
.
rec_model_dir
,
rec_url
=
confirm_model_dir_url
(
params
.
rec_model_dir
,
params
.
rec_model_dir
,
os
.
path
.
join
(
BASE_DIR
,
VERSION
,
'ocr'
,
'rec'
,
lang
),
os
.
path
.
join
(
BASE_DIR
,
'whl'
,
'rec'
,
lang
),
rec_model_config
[
'url'
])
rec_model_config
[
'url'
])
table_model_config
=
get_model_config
(
table_model_config
=
get_model_config
(
'STRUCTURE'
,
params
.
structure_version
,
'table'
,
'en'
)
'STRUCTURE'
,
params
.
structure_version
,
'table'
,
'en'
)
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
,
VERSION
,
'ocr'
,
'table'
),
os
.
path
.
join
(
BASE_DIR
,
'whl'
,
'table'
),
table_model_config
[
'url'
])
table_model_config
[
'url'
])
# download model
# download model
maybe_download
(
params
.
det_model_dir
,
det_url
)
maybe_download
(
params
.
det_model_dir
,
det_url
)
maybe_download
(
params
.
rec_model_dir
,
rec_url
)
maybe_download
(
params
.
rec_model_dir
,
rec_url
)
...
...
tools/infer/utility.py
View file @
f85b3ea8
...
@@ -622,7 +622,6 @@ def get_rotate_crop_image(img, points):
...
@@ -622,7 +622,6 @@ def get_rotate_crop_image(img, points):
def
check_gpu
(
use_gpu
):
def
check_gpu
(
use_gpu
):
if
use_gpu
and
not
paddle
.
is_compiled_with_cuda
():
if
use_gpu
and
not
paddle
.
is_compiled_with_cuda
():
use_gpu
=
False
use_gpu
=
False
return
use_gpu
return
use_gpu
...
...
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