diff options
Diffstat (limited to 'tensorflow/python/keras/applications/mobilenet.py')
| -rw-r--r-- | tensorflow/python/keras/applications/mobilenet.py | 25 |
1 files changed, 19 insertions, 6 deletions
diff --git a/tensorflow/python/keras/applications/mobilenet.py b/tensorflow/python/keras/applications/mobilenet.py index 3528f027b3..1f71a5ae99 100644 --- a/tensorflow/python/keras/applications/mobilenet.py +++ b/tensorflow/python/keras/applications/mobilenet.py @@ -20,12 +20,25 @@ from __future__ import division from __future__ import print_function from keras_applications import mobilenet + +from tensorflow.python.keras.applications import keras_modules_injection from tensorflow.python.util.tf_export import tf_export -MobileNet = mobilenet.MobileNet -decode_predictions = mobilenet.decode_predictions -preprocess_input = mobilenet.preprocess_input -tf_export('keras.applications.mobilenet.MobileNet', - 'keras.applications.MobileNet')(MobileNet) -tf_export('keras.applications.mobilenet.preprocess_input')(preprocess_input) +@tf_export('keras.applications.mobilenet.MobileNet', + 'keras.applications.MobileNet') +@keras_modules_injection +def MobileNet(*args, **kwargs): + return mobilenet.MobileNet(*args, **kwargs) + + +@tf_export('keras.applications.mobilenet.decode_predictions') +@keras_modules_injection +def decode_predictions(*args, **kwargs): + return mobilenet.decode_predictions(*args, **kwargs) + + +@tf_export('keras.applications.mobilenet.preprocess_input') +@keras_modules_injection +def preprocess_input(*args, **kwargs): + return mobilenet.preprocess_input(*args, **kwargs) |