Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
Paddle
提交
bc48453b
P
Paddle
项目概览
BaiXuePrincess
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
bc48453b
编写于
4月 30, 2019
作者:
X
xiaoting
提交者:
Kaipeng Deng
4月 30, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
polish the label_smooth (#17138)
* polish the label_smooth test=develop * polish code test=develop
上级
bf4b21fa
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
9 addition
and
6 deletion
+9
-6
paddle/fluid/operators/detection/yolov3_loss_op.h
paddle/fluid/operators/detection/yolov3_loss_op.h
+6
-4
python/paddle/fluid/tests/unittests/test_yolov3_loss_op.py
python/paddle/fluid/tests/unittests/test_yolov3_loss_op.py
+3
-2
未找到文件。
paddle/fluid/operators/detection/yolov3_loss_op.h
浏览文件 @
bc48453b
...
...
@@ -282,8 +282,9 @@ class Yolov3LossKernel : public framework::OpKernel<T> {
T
label_pos
=
1.0
;
T
label_neg
=
0.0
;
if
(
use_label_smooth
)
{
label_pos
=
1.0
-
1.0
/
static_cast
<
T
>
(
class_num
);
label_neg
=
1.0
/
static_cast
<
T
>
(
class_num
);
T
smooth_weight
=
std
::
min
(
1.0
/
static_cast
<
T
>
(
class_num
),
1.0
/
40
);
label_pos
=
1.0
-
smooth_weight
;
label_neg
=
smooth_weight
;
}
const
T
*
input_data
=
input
->
data
<
T
>
();
...
...
@@ -437,8 +438,9 @@ class Yolov3LossGradKernel : public framework::OpKernel<T> {
T
label_pos
=
1.0
;
T
label_neg
=
0.0
;
if
(
use_label_smooth
)
{
label_pos
=
1.0
-
1.0
/
static_cast
<
T
>
(
class_num
);
label_neg
=
1.0
/
static_cast
<
T
>
(
class_num
);
T
smooth_weight
=
std
::
min
(
1.0
/
static_cast
<
T
>
(
class_num
),
1.0
/
40
);
label_pos
=
1.0
-
smooth_weight
;
label_neg
=
smooth_weight
;
}
const
T
*
input_data
=
input
->
data
<
T
>
();
...
...
python/paddle/fluid/tests/unittests/test_yolov3_loss_op.py
浏览文件 @
bc48453b
...
...
@@ -81,8 +81,9 @@ def YOLOv3Loss(x, gtbox, gtlabel, gtscore, attrs):
x
=
x
.
reshape
((
n
,
mask_num
,
5
+
class_num
,
h
,
w
)).
transpose
((
0
,
1
,
3
,
4
,
2
))
loss
=
np
.
zeros
((
n
)).
astype
(
'float32'
)
label_pos
=
1.0
-
1.0
/
class_num
if
use_label_smooth
else
1.0
label_neg
=
1.0
/
class_num
if
use_label_smooth
else
0.0
smooth_weight
=
min
(
1.0
/
class_num
,
1.0
/
40
)
label_pos
=
1.0
-
smooth_weight
if
use_label_smooth
else
1.0
label_neg
=
smooth_weight
if
use_label_smooth
else
0.0
pred_box
=
x
[:,
:,
:,
:,
:
4
].
copy
()
grid_x
=
np
.
tile
(
np
.
arange
(
w
).
reshape
((
1
,
w
)),
(
h
,
1
))
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录