Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
聚火科技码农
PaddleOCR
提交
6081c67b
P
PaddleOCR
项目概览
聚火科技码农
/
PaddleOCR
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleOCR
通知
1
Star
0
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看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
6081c67b
编写于
5月 25, 2020
作者:
D
Double_V
提交者:
GitHub
5月 25, 2020
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #95 from LDOUBLEV/fixocr
fix predict_det not found unclip_ratio
上级
f47165ba
8a6f1062
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
13 addition
and
4 deletion
+13
-4
README.md
README.md
+2
-2
doc/detection.md
doc/detection.md
+7
-0
tools/infer/predict_det.py
tools/infer/predict_det.py
+3
-2
tools/infer/utility.py
tools/infer/utility.py
+1
-0
未找到文件。
README.md
浏览文件 @
6081c67b
...
...
@@ -64,13 +64,13 @@ PaddleOCR开源的文本检测算法列表:
在ICDAR2015文本检测公开数据集上,算法效果如下:
|模型|骨干网络|precision|recall|Hmean|下载链接|
|-|-|-|-|
|-|-|-|-|
-|-|
|EAST|ResNet50_vd|88.18%|85.51|86.82%|
[
下载链接
](
https://paddleocr.bj.bcebos.com/det_r50_vd_east.tar
)
|
|EAST|MobileNetV3|81.67%|79.83%|80.74%|
[
下载链接
](
https://paddleocr.bj.bcebos.com/det_mv3_east.tar
)
|
|DB|ResNet50_vd|83.79%|80.65%|82.19%|
[
下载链接
](
https://paddleocr.bj.bcebos.com/det_r50_vd_db.tar
)
|
|DB|MobileNetV3|75.92%|73.18%|74.53%|
[
下载链接
](
https://paddleocr.bj.bcebos.com/det_mv3_db.tar
)
|
*
注: 上述
模型的训练和评估,
设置后处理参数box_thresh=0.6,unclip_ratio=1.5,使用不同数据集、不同模型训练,可调整这两个参数进行优化
*
注: 上述
DB模型的训练和评估,需
设置后处理参数box_thresh=0.6,unclip_ratio=1.5,使用不同数据集、不同模型训练,可调整这两个参数进行优化
PaddleOCR文本检测算法的训练和使用请参考文档教程中
[
文本检测模型训练/评估/预测
](
./doc/detection.md
)
。
...
...
doc/detection.md
浏览文件 @
6081c67b
...
...
@@ -75,6 +75,7 @@ python3 tools/eval.py -c configs/det/det_mv3_db.yml -o Global.checkpoints="{pat
python3 tools/eval.py -c configs/det/det_mv3_db.yml -o Global.checkpoints="./output/det_db/best_accuracy" PostProcess.box_thresh=0.6 PostProcess.unclip_ratio=1.5
```
*
注:box_thresh、unclip_ratio是DB后处理所需要的参数,在评估EAST模型时不需要设置
## 测试检测效果
...
...
@@ -83,6 +84,12 @@ python3 tools/eval.py -c configs/det/det_mv3_db.yml -o Global.checkpoints="./ou
python3 tools/infer_det.py -c configs/det/det_mv3_db.yml -o TestReader.infer_img="./doc/imgs_en/img_10.jpg" Global.checkpoints="./output/det_db/best_accuracy"
```
测试DB模型时,调整后处理阈值,
```
python3 tools/infer_det.py -c configs/det/det_mv3_db.yml -o TestReader.infer_img="./doc/imgs_en/img_10.jpg" Global.checkpoints="./output/det_db/best_accuracy" PostProcess.box_thresh=0.6 PostProcess.unclip_ratio=1.5
```
测试文件夹下所有图像的检测效果
```
python3 tools/infer_det.py -c configs/det/det_mv3_db.yml -o TestReader.infer_img="./doc/imgs_en/" Global.checkpoints="./output/det_db/best_accuracy"
...
...
tools/infer/predict_det.py
浏览文件 @
6081c67b
...
...
@@ -39,6 +39,7 @@ class TextDetector(object):
postprocess_params
[
"thresh"
]
=
args
.
det_db_thresh
postprocess_params
[
"box_thresh"
]
=
args
.
det_db_box_thresh
postprocess_params
[
"max_candidates"
]
=
1000
postprocess_params
[
"unclip_ratio"
]
=
args
.
det_db_unclip_ratio
self
.
postprocess_op
=
DBPostProcess
(
postprocess_params
)
elif
self
.
det_algorithm
==
"EAST"
:
self
.
preprocess_op
=
EASTProcessTest
(
preprocess_params
)
...
...
@@ -142,5 +143,5 @@ if __name__ == "__main__":
src_im
=
utility
.
draw_text_det_res
(
dt_boxes
,
image_file
)
img_name_pure
=
image_file
.
split
(
"/"
)[
-
1
]
cv2
.
imwrite
(
"./inference_results/det_res_%s"
%
img_name_pure
,
src_im
)
if
count
>
1
:
print
(
"Avg Time:"
,
total_time
/
(
count
-
1
))
if
count
>
1
:
print
(
"Avg Time:"
,
total_time
/
(
count
-
1
))
tools/infer/utility.py
浏览文件 @
6081c67b
...
...
@@ -45,6 +45,7 @@ def parse_args():
#DB parmas
parser
.
add_argument
(
"--det_db_thresh"
,
type
=
float
,
default
=
0.3
)
parser
.
add_argument
(
"--det_db_box_thresh"
,
type
=
float
,
default
=
0.5
)
parser
.
add_argument
(
"--det_db_unclip_ratio"
,
type
=
float
,
default
=
2.0
)
#EAST parmas
parser
.
add_argument
(
"--det_east_score_thresh"
,
type
=
float
,
default
=
0.8
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录