Fix issue with MobileNetV3 in Keras

tensorflow/tensorflow#40217 keras-team/keras-applications#183
woctezuma · Aug 22, 2020 · f15c829 · f15c829
1 parent f5e3685
commit f15c829
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/model_utils.py b/model_utils.py
@@ -1,6 +1,6 @@
 import numpy as np
-from keras.applications.mobilenet_v3 import MobileNetV3Small
-from keras.applications.mobilenet_v3 import preprocess_input
+from tensorflow.keras.applications import MobileNetV3Small
+from tensorflow.keras.applications.mobilenet_v3 import preprocess_input
 from keras.preprocessing.image import img_to_array
 
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
+tf-nightly
 cython==0.29.21
 numpy==1.19.1
 keras==2.4.3