Fix Tf export for BottleneckCSP (ultralytics#7330)

fhkiel-mlaip · Apr 7, 2022 · fb3a952 · fb3a952
1 parent 57447c0
commit fb3a952
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):