Add relu6 (keras-team#10521)

AlexMikhalev · Jun 25, 2018 · 37b1645 · 37b1645
1 parent ba4aac6
commit 37b1645
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/keras/applications/mobilenet.py b/keras/applications/mobilenet.py
@@ -4,6 +4,7 @@
 
 from keras_applications import mobilenet
 
+relu6 = mobilenet.relu6
 MobileNet = mobilenet.MobileNet
 decode_predictions = mobilenet.decode_predictions
 preprocess_input = mobilenet.preprocess_input
diff --git a/keras/applications/mobilenetv2.py b/keras/applications/mobilenetv2.py
@@ -4,6 +4,7 @@
 
 from keras_applications import mobilenet_v2
 
+relu6 = mobilenet_v2.relu6
 MobileNetV2 = mobilenet_v2.MobileNetV2
 decode_predictions = mobilenet_v2.decode_predictions
 preprocess_input = mobilenet_v2.preprocess_input