Unverified 提交 b7faeda0 authored 作者: Nrupatunga's avatar Nrupatunga 提交者: GitHub

Fix Tf export for BottleneckCSP (#7330)

上级 0ca85ed6
...@@ -152,7 +152,7 @@ class TFBottleneckCSP(keras.layers.Layer): ...@@ -152,7 +152,7 @@ class TFBottleneckCSP(keras.layers.Layer):
self.cv3 = TFConv2d(c_, c_, 1, 1, bias=False, w=w.cv3) self.cv3 = TFConv2d(c_, c_, 1, 1, bias=False, w=w.cv3)
self.cv4 = TFConv(2 * c_, c2, 1, 1, w=w.cv4) self.cv4 = TFConv(2 * c_, c2, 1, 1, w=w.cv4)
self.bn = TFBN(w.bn) self.bn = TFBN(w.bn)
self.act = lambda x: keras.activations.relu(x, alpha=0.1) self.act = lambda x: keras.activations.swish(x)
self.m = keras.Sequential([TFBottleneck(c_, c_, shortcut, g, e=1.0, w=w.m[j]) for j in range(n)]) self.m = keras.Sequential([TFBottleneck(c_, c_, shortcut, g, e=1.0, w=w.m[j]) for j in range(n)])
def call(self, inputs): def call(self, inputs):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论