Skip to content

P
PaddleClas

PaddlePaddle / PaddleClas
1 年多 前同步成功

通知 115
Star 4999
Fork 1114
P
PaddleClas
提交 1153ad65 编写于 作者: W weishengyu
浏览文件
操作

dbg

上级 296294b9
隐藏空白更改
内联 并排
Showing with 8 addition and 12 deletion
ppcls/arch/backbone/legendary_models/hrnet.py
浏览文件 @ 1153ad65
...@@ -496,16 +496,14 @@ class HRNet(TheseusLayer): ...@@ -496,16 +496,14 @@ class HRNet(TheseusLayer):
self.la1 = Layer1(num_channels=64, has_se=has_se, name="layer2") self.la1 = Layer1(num_channels=64, has_se=has_se, name="layer2")
self.tr1_1 = BasicBlock( self.tr1_1 = ConvBNLayer(
num_channels=256, num_channels=256,
num_filters=width, num_filters=width,
has_se=has_se, filter_size=3)
name="tr1_1") self.tr1_2 = ConvBNLayer(
self.tr1_2 = BasicBlock(
num_channels=256, num_channels=256,
num_filters=width * 2, num_filters=width * 2,
has_se=has_se, filter_size=3)
name="tr1_2")
self.st2 = Stage( self.st2 = Stage(
num_channels=channels_2, num_channels=channels_2,
...@@ -514,11 +512,10 @@ class HRNet(TheseusLayer): ...@@ -514,11 +512,10 @@ class HRNet(TheseusLayer):
has_se=self.has_se, has_se=self.has_se,
name="st2") name="st2")
self.tr2 = BasicBlock( self.tr2 = ConvBNLayer(
num_channels=width * 2, num_channels=width * 2,
num_filters=width * 4, num_filters=width * 4,
has_se=has_se, filter_size=3)
name="tr2")
self.st3 = Stage( self.st3 = Stage(
num_channels=channels_3, num_channels=channels_3,
num_modules=num_modules_3, num_modules=num_modules_3,
...@@ -526,11 +523,10 @@ class HRNet(TheseusLayer): ...@@ -526,11 +523,10 @@ class HRNet(TheseusLayer):
has_se=self.has_se, has_se=self.has_se,
name="st3") name="st3")
self.tr3 = BasicBlock( self.tr3 = ConvBNLayer(
num_channels=width * 4, num_channels=width * 4,
num_filters=width * 8, num_filters=width * 8,
has_se=has_se, filter_size=3)
name="tr3")
self.st4 = Stage( self.st4 = Stage(
num_channels=channels_4, num_channels=channels_4,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或