Fix keras imports

contrib/tf_serving/save_tfs_model.py

@@ -1,7 +1,7 @@
-import keras.backend as K
+import tensorflow.keras.backend as K
 import argparse
-from keras.applications.mobilenet import DepthwiseConv2D, relu6
-from keras.utils.generic_utils import CustomObjectScope
+from tensorflow.keras.applications.mobilenet import DepthwiseConv2D, relu6
+from tensorflow.keras.utils.generic_utils import CustomObjectScope
 from tensorflow.python.saved_model import builder as saved_model_builder
 from tensorflow.python.saved_model import tag_constants
 from tensorflow.python.saved_model.signature_def_utils_impl import \

contrib/tf_serving/tfs_sample_client.py

@@ -1,6 +1,6 @@
 import json
 import argparse
-import keras
+import tensorflow.keras as keras
 import numpy as np
 import tensorflow as tf
 from src.utils import utils

src/handlers/data_generator.py

@@ -1,7 +1,7 @@
 
 import os
 import numpy as np
-import keras
+import tensorflow.keras as keras
 from utils import utils
 
 

src/handlers/model_builder.py

@@ -1,8 +1,8 @@
 
 import importlib
-from keras.models import Model
-from keras.layers import Dropout, Dense
-from keras.optimizers import Adam
+from tensorflow.keras.models import Model
+from tensorflow.keras.layers import Dropout, Dense
+from tensorflow.keras.optimizers import Adam
 from utils.losses import earth_movers_distance
 
 
@@ -21,11 +21,11 @@ def __init__(self, base_model_name, n_classes=10, learning_rate=0.001, dropout_r
     def _get_base_module(self):
         # import Keras base model module
         if self.base_model_name == 'InceptionV3':
-            self.base_module = importlib.import_module('keras.applications.inception_v3')
+            self.base_module = importlib.import_module('tensorflow.keras.applications.inception_v3')
         elif self.base_model_name == 'InceptionResNetV2':
-            self.base_module = importlib.import_module('keras.applications.inception_resnet_v2')
+            self.base_module = importlib.import_module('tensorflow.keras.applications.inception_resnet_v2')
         else:
-            self.base_module = importlib.import_module('keras.applications.'+self.base_model_name.lower())
+            self.base_module = importlib.import_module('tensorflow.keras.applications.'+self.base_model_name.lower())
 
     def build(self):
         # get base model class

src/trainer/train.py

@@ -1,8 +1,8 @@
 
 import os
 import argparse
-from keras import backend as K
-from keras.callbacks import ModelCheckpoint
+from tensorflow.keras import backend as K
+from tensorflow.keras.callbacks import ModelCheckpoint
 from sklearn.model_selection import train_test_split
 from handlers.data_generator import TrainDataGenerator, TestDataGenerator
 from handlers.model_builder import Nima

src/utils/keras_utils.py

@@ -1,6 +1,6 @@
 
-import keras.backend as K
-from keras.callbacks import TensorBoard
+import tensorflow.keras.backend as K
+from tensorflow.keras.callbacks import TensorBoard
 
 
 class TensorBoardBatch(TensorBoard):

src/utils/losses.py

@@ -1,5 +1,5 @@
 
-from keras import backend as K
+from tensorflow.keras import backend as K
 
 
 def earth_movers_distance(y_true, y_pred):

src/utils/utils.py

@@ -1,7 +1,7 @@
 
 import os
 import json
-import keras
+import tensorflow.keras as keras
 import numpy as np