未验证 提交 155190c2 编写于 作者: F Feng Ni 提交者: GitHub

[MOT] fix fairmot centernet_head bias init (#4206)

上级 fe6b9a74
...@@ -98,7 +98,8 @@ class CenterNetHead(nn.Layer): ...@@ -98,7 +98,8 @@ class CenterNetHead(nn.Layer):
stride=1, stride=1,
padding=0, padding=0,
bias=True)) bias=True))
self.heatmap[2].conv.bias[:] = -2.19 with paddle.no_grad():
self.heatmap[2].conv.bias[:] = -2.19
self.size = nn.Sequential( self.size = nn.Sequential(
ConvLayer( ConvLayer(
in_channels, head_planes, kernel_size=3, padding=1, bias=True), in_channels, head_planes, kernel_size=3, padding=1, bias=True),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册