Fix bug in relu6 activation #38

MarsTechHAN · Jan 28, 2022 · d0257c8 · d0257c8
1 parent 96d7f1b
commit d0257c8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/keras2ncnn/keras_converter.py b/keras2ncnn/keras_converter.py
@@ -570,7 +570,8 @@ def Activation_helper(
             else:
                 negative_slope = 0.0
 
-            if '_relu6' in layer['layer']['config']['activation']:
+            if '_relu6' in layer['layer']['config']['activation'] or \
+                    'relu6' in layer['layer']['config']['activation']:
                 layer['layer']['config']['max_value'] = 6.0
                 layer['layer']['config']['activation'] = 'relu'