Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
a2575523
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
大约 1 年 前同步成功
通知
282
Star
12117
Fork
2091
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
200
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleHub
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
200
Issue
200
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
a2575523
编写于
4月 28, 2020
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update param name in detection task
上级
790860aa
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
10 addition
and
12 deletion
+10
-12
paddlehub/finetune/task/faster_rcnn_task.py
paddlehub/finetune/task/faster_rcnn_task.py
+8
-10
paddlehub/finetune/task/yolo_task.py
paddlehub/finetune/task/yolo_task.py
+2
-2
未找到文件。
paddlehub/finetune/task/faster_rcnn_task.py
浏览文件 @
a2575523
...
...
@@ -67,30 +67,28 @@ class FasterRCNNTask(DetectionTask):
head_feat
=
self
.
main_program
.
global_block
().
vars
[
self
.
predict_feature
[
0
].
name
]
# Rename following layers for: ValueError: Variable cls_score_w has been created before.
# the previous shape is (2048, 81); the new shape is (100352, 81).
# They are not matched.
cls_score
=
fluid
.
layers
.
fc
(
input
=
head_feat
,
size
=
self
.
num_classes
,
act
=
None
,
name
=
'
my
_cls_score'
,
name
=
'
paddlehub_rcnn
_cls_score'
,
param_attr
=
ParamAttr
(
name
=
'my_cls_score_w'
,
initializer
=
Normal
(
loc
=
0.0
,
scale
=
0.01
)),
name
=
'paddlehub_rcnn_cls_score_weights'
,
initializer
=
Normal
(
loc
=
0.0
,
scale
=
0.01
)),
bias_attr
=
ParamAttr
(
name
=
'
my_cls_score_b
'
,
name
=
'
paddlehub_rcnn_cls_score_bias
'
,
learning_rate
=
2.
,
regularizer
=
L2Decay
(
0.
)))
bbox_pred
=
fluid
.
layers
.
fc
(
input
=
head_feat
,
size
=
4
*
self
.
num_classes
,
act
=
None
,
name
=
'
my
_bbox_pred'
,
name
=
'
paddlehub_rcnn
_bbox_pred'
,
param_attr
=
ParamAttr
(
name
=
'
my_bbox_pred_w'
,
initializer
=
Normal
(
loc
=
0.0
,
scale
=
0.001
)),
name
=
'
paddlehub_rcnn_bbox_pred_weights'
,
initializer
=
Normal
(
loc
=
0.0
,
scale
=
0.001
)),
bias_attr
=
ParamAttr
(
name
=
'
my_bbox_pred_b
'
,
name
=
'
paddlehub_rcnn_bbox_pred_bias
'
,
learning_rate
=
2.
,
regularizer
=
L2Decay
(
0.
)))
...
...
paddlehub/finetune/task/yolo_task.py
浏览文件 @
a2575523
...
...
@@ -90,10 +90,10 @@ class YOLOTask(DetectionTask):
act
=
None
,
# Rename for: conflict with module pretrain weights
param_attr
=
ParamAttr
(
name
=
"
ft
_yolo_output.{}.conv.weights"
.
format
(
i
)),
name
=
"
paddlehub
_yolo_output.{}.conv.weights"
.
format
(
i
)),
bias_attr
=
ParamAttr
(
regularizer
=
L2Decay
(
0.
),
name
=
"
ft
_yolo_output.{}.conv.bias"
.
format
(
i
)))
name
=
"
paddlehub
_yolo_output.{}.conv.bias"
.
format
(
i
)))
outputs
.
append
(
block_out
)
if
self
.
is_train_phase
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录