Fix Tf export for BottleneckCSP (#7330)

zhongcaoren95 · Apr 7, 2022 · b7faeda · b7faeda
1 parent 0ca85ed
commit b7faeda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/tf.py b/models/tf.py
@@ -152,7 +152,7 @@ def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5, w=None):
         self.cv3 = TFConv2d(c_, c_, 1, 1, bias=False, w=w.cv3)
         self.cv4 = TFConv(2 * c_, c2, 1, 1, w=w.cv4)
         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)])
 
     def call(self, inputs):