Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
e0847f18
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
e0847f18
编写于
2月 20, 2023
作者:
T
tianyi1997
提交者:
HydrogenSulfate
2月 28, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update pretrained backbone
上级
f313a6d8
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
7 addition
and
6 deletion
+7
-6
ppcls/arch/backbone/variant_models/resnet_variant.py
ppcls/arch/backbone/variant_models/resnet_variant.py
+6
-5
ppcls/configs/reid/MetaBIN_ResNet50_cross_domain.yaml
ppcls/configs/reid/MetaBIN_ResNet50_cross_domain.yaml
+1
-1
未找到文件。
ppcls/arch/backbone/variant_models/resnet_variant.py
浏览文件 @
e0847f18
...
...
@@ -175,6 +175,10 @@ def ResNet50_metabin(pretrained=False,
def
bn2metabin
(
bn
,
pattern
):
metabin
=
MetaBIN
(
bn
.
weight
.
shape
[
0
])
metabin
.
batch_norm
.
weight
.
set_value
(
bn
.
weight
)
metabin
.
batch_norm
.
bias
.
set_value
(
bn
.
bias
)
metabin
.
batch_norm
.
_variance
.
set_value
(
bn
.
_variance
)
metabin
.
batch_norm
.
_mean
.
set_value
(
bn
.
_mean
)
return
metabin
def
setup_optimize_attr
(
model
,
bias_lr_factor
):
...
...
@@ -184,19 +188,16 @@ def ResNet50_metabin(pretrained=False,
if
"bias"
in
name
:
params
.
optimize_attr
[
'learning_rate'
]
=
bias_lr_factor
stride_list
=
[
2
,
2
,
2
,
2
,
1
]
pattern
=
[]
pattern
.
extend
([
"blocks[{}].conv{}.bn"
.
format
(
i
,
j
)
\
for
i
in
range
(
16
)
for
j
in
range
(
3
)])
pattern
.
extend
([
"blocks[{}].short.bn"
.
format
(
i
)
for
i
in
[
0
,
3
,
7
,
13
]])
pattern
.
append
(
"stem[0].bn"
)
model
=
ResNet50
(
pretrained
=
False
,
use_ssld
=
use_ssld
,
stride_list
=
stride_list
,
**
kwargs
)
model
=
ResNet50
_last_stage_stride1
(
pretrained
=
pretrained
,
use_ssld
=
use_ssld
,
**
kwargs
)
model
.
upgrade_sublayer
(
pattern
,
bn2metabin
)
setup_optimize_attr
(
model
=
model
,
bias_lr_factor
=
bias_lr_factor
)
_load_pretrained
(
pretrained
,
model
,
MODEL_URLS
[
"ResNet50"
],
use_ssld
)
return
model
ppcls/configs/reid/MetaBIN_ResNet50_cross_domain.yaml
浏览文件 @
e0847f18
...
...
@@ -29,7 +29,7 @@ Arch:
name
:
"
RecModel"
Backbone
:
name
:
"
ResNet50_metabin"
pretrained
:
False
pretrained
:
https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/others/resnet50-19c8e357_torch2paddle.pdparams
bias_lr_factor
:
2.0
BackboneStopLayer
:
name
:
"
flatten"
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录