Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleX
提交
f55d7014
P
PaddleX
项目概览
PaddlePaddle
/
PaddleX
通知
138
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
43
列表
看板
标记
里程碑
合并请求
5
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleX
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
43
Issue
43
列表
看板
标记
里程碑
合并请求
5
合并请求
5
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
f55d7014
编写于
5月 11, 2020
作者:
C
Channingss
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
optimize code structure
上级
0611fe1a
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
18 addition
and
9 deletion
+18
-9
paddlex/cv/transforms/det_transforms.py
paddlex/cv/transforms/det_transforms.py
+18
-9
未找到文件。
paddlex/cv/transforms/det_transforms.py
浏览文件 @
f55d7014
...
...
@@ -212,11 +212,25 @@ class Padding:
Args:
coarsest_stride (int): 填充后的图像长、宽为该参数的倍数,默认为1。
target_size (int|list): 填充后的图像长、宽,默认为None,coarset_stride优先级更高。
target_size (int|list|tuple): 填充后的图像长、宽,默认为None,coarset_stride优先级更高。
Raises:
TypeError: 形参`target_size`数据类型不满足需求。
ValueError: 形参`target_size`为(list|tuple)时,长度不满足需求。
"""
def
__init__
(
self
,
coarsest_stride
=
1
,
target_size
=
None
):
self
.
coarsest_stride
=
coarsest_stride
if
target_size
is
not
None
:
if
not
isinstance
(
target_size
,
int
):
if
not
isinstance
(
target_size
,
tuple
)
and
not
isinstance
(
target_size
,
list
):
raise
TypeError
(
"Padding: Type of target_size must in (int|list|tuple)."
)
elif
len
(
target_size
)
!=
2
:
raise
ValueError
(
"Padding: Length of target_size must equal 2."
)
self
.
target_size
=
target_size
def
__call__
(
self
,
im
,
im_info
=
None
,
label_info
=
None
):
...
...
@@ -237,12 +251,6 @@ class Padding:
ValueError: coarsest_stride,target_size需有且只有一个被指定。
ValueError: target_size小于原图的大小。
"""
if
self
.
coarsest_stride
==
1
and
self
.
target_size
is
None
:
if
label_info
is
None
:
return
(
im
,
im_info
)
else
:
return
(
im
,
im_info
,
label_info
)
if
im_info
is
None
:
im_info
=
dict
()
if
not
isinstance
(
im
,
np
.
ndarray
):
...
...
@@ -254,10 +262,11 @@ class Padding:
if
isinstance
(
self
.
target_size
,
int
):
padding_im_h
=
self
.
target_size
padding_im_w
=
self
.
target_size
elif
isinstance
(
self
.
target_size
,
list
):
elif
isinstance
(
self
.
target_size
,
list
)
or
isinstance
(
self
.
target_size
,
tuple
):
padding_im_w
=
self
.
target_size
[
0
]
padding_im_h
=
self
.
target_size
[
1
]
elif
self
.
coarsest_stride
>
1
:
elif
self
.
coarsest_stride
>
0
:
padding_im_h
=
int
(
np
.
ceil
(
im_h
/
self
.
coarsest_stride
)
*
self
.
coarsest_stride
)
padding_im_w
=
int
(
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录