Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleOCR
提交
62a9e8df
P
PaddleOCR
项目概览
s920243400
/
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看板
提交
62a9e8df
编写于
6月 22, 2022
作者:
L
LDOUBLEV
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix
上级
a24e30a7
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
18 addition
and
32 deletion
+18
-32
ppocr/data/imaug/label_ops.py
ppocr/data/imaug/label_ops.py
+9
-16
tools/infer/predict_det.py
tools/infer/predict_det.py
+9
-16
未找到文件。
ppocr/data/imaug/label_ops.py
浏览文件 @
62a9e8df
...
@@ -70,22 +70,15 @@ class DetLabelEncode(object):
...
@@ -70,22 +70,15 @@ class DetLabelEncode(object):
return
data
return
data
def
order_points_clockwise
(
self
,
pts
):
def
order_points_clockwise
(
self
,
pts
):
"""
rect
=
np
.
zeros
((
4
,
2
),
dtype
=
"float32"
)
refer to :https://github.com/PyImageSearch/imutils/blob/9f740a53bcc2ed7eba2558afed8b4c17fd8a1d4c/imutils/perspective.py#L9
s
=
pts
.
sum
(
axis
=
1
)
"""
rect
[
0
]
=
pts
[
np
.
argmin
(
s
)]
# sort the points based on their x-coordinates
rect
[
2
]
=
pts
[
np
.
argmax
(
s
)]
xSorted
=
pts
[
np
.
argsort
(
pts
[:,
0
]),
:]
tmp
=
np
.
delete
(
pts
,
(
np
.
argmin
(
s
),
np
.
argmax
(
s
)),
axis
=
0
)
diff
=
np
.
diff
(
np
.
array
(
tmp
),
axis
=
1
)
leftMost
=
xSorted
[:
2
,
:]
rect
[
1
]
=
tmp
[
np
.
argmin
(
diff
)]
rightMost
=
xSorted
[
2
:,
:]
rect
[
3
]
=
tmp
[
np
.
argmax
(
diff
)]
return
rect
leftMost
=
leftMost
[
np
.
argsort
(
leftMost
[:,
1
]),
:]
(
tl
,
bl
)
=
leftMost
D
=
dist
.
cdist
(
tl
[
np
.
newaxis
],
rightMost
,
"euclidean"
)[
0
]
(
br
,
tr
)
=
rightMost
[
np
.
argsort
(
D
)[::
-
1
],
:]
return
np
.
array
([
tl
,
tr
,
br
,
bl
],
dtype
=
"float32"
)
def
expand_points_num
(
self
,
boxes
):
def
expand_points_num
(
self
,
boxes
):
max_points_num
=
0
max_points_num
=
0
...
...
tools/infer/predict_det.py
浏览文件 @
62a9e8df
...
@@ -151,22 +151,15 @@ class TextDetector(object):
...
@@ -151,22 +151,15 @@ class TextDetector(object):
logger
=
logger
)
logger
=
logger
)
def
order_points_clockwise
(
self
,
pts
):
def
order_points_clockwise
(
self
,
pts
):
"""
rect
=
np
.
zeros
((
4
,
2
),
dtype
=
"float32"
)
refer to :https://github.com/PyImageSearch/imutils/blob/9f740a53bcc2ed7eba2558afed8b4c17fd8a1d4c/imutils/perspective.py#L9
s
=
pts
.
sum
(
axis
=
1
)
"""
rect
[
0
]
=
pts
[
np
.
argmin
(
s
)]
# sort the points based on their x-coordinates
rect
[
2
]
=
pts
[
np
.
argmax
(
s
)]
xSorted
=
pts
[
np
.
argsort
(
pts
[:,
0
]),
:]
tmp
=
np
.
delete
(
pts
,
(
np
.
argmin
(
s
),
np
.
argmax
(
s
)),
axis
=
0
)
diff
=
np
.
diff
(
np
.
array
(
tmp
),
axis
=
1
)
leftMost
=
xSorted
[:
2
,
:]
rect
[
1
]
=
tmp
[
np
.
argmin
(
diff
)]
rightMost
=
xSorted
[
2
:,
:]
rect
[
3
]
=
tmp
[
np
.
argmax
(
diff
)]
return
rect
leftMost
=
leftMost
[
np
.
argsort
(
leftMost
[:,
1
]),
:]
(
tl
,
bl
)
=
leftMost
D
=
dist
.
cdist
(
tl
[
np
.
newaxis
],
rightMost
,
"euclidean"
)[
0
]
(
br
,
tr
)
=
rightMost
[
np
.
argsort
(
D
)[::
-
1
],
:]
return
np
.
array
([
tl
,
tr
,
br
,
bl
],
dtype
=
"float32"
)
def
clip_det_res
(
self
,
points
,
img_height
,
img_width
):
def
clip_det_res
(
self
,
points
,
img_height
,
img_width
):
for
pno
in
range
(
points
.
shape
[
0
]):
for
pno
in
range
(
points
.
shape
[
0
]):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录