Replace keras with tf.keras

src/nets/Loss.py

@@ -1,6 +1,6 @@
 import os
-from keras import backend as K
 import tensorflow as tf
+from tf.keras import backend as K
 import numpy as np
 
 def multi_category_focal_loss2(gamma=2., alpha=1):

src/nets/mobilenet.py

@@ -1,8 +1,8 @@
-from keras import backend as K
-from keras.layers import (Activation, BatchNormalization, Conv2D,
+from tensorflow.keras import backend as K
+from tensorflow.keras.layers import (Activation, BatchNormalization, Conv2D,
                           DepthwiseConv2D, Dropout, GlobalAveragePooling2D,
                           Input, Reshape)
-from keras.models import Model
+from tensorflow.keras.models import Model
 
 
 def _conv_block(inputs, filters, alpha, kernel=(3, 3), strides=(1, 1)):

src/nets/resnet50.py

@@ -1,8 +1,8 @@
-from keras import layers
-from keras.layers import (Activation, AveragePooling2D, BatchNormalization,
+from tensorflow.keras import layers
+from tensorflow.keras.layers import (Activation, AveragePooling2D, BatchNormalization,
                           Conv2D, Dense, Flatten, Input, MaxPooling2D,
                           ZeroPadding2D)
-from keras.models import Model
+from tensorflow.keras.models import Model
 
 
 def identity_block(input_tensor, kernel_size, filters, stage, block):

src/nets/vgg16.py

@@ -1,5 +1,5 @@
-from keras.layers import Conv2D, Dense, Flatten, Input, MaxPooling2D
-from keras.models import Model        #导入包Conv2D是卷积核 Flatten是展开 Input输入  MaxPooling2D最大卷积核
+from tensorflow.keras.layers import Conv2D, Dense, Flatten, Input, MaxPooling2D
+from tensorflow.keras.models import Model        #导入包Conv2D是卷积核 Flatten是展开 Input输入  MaxPooling2D最大卷积核
 
 
 def VGG16(input_shape=None, classes=1000): #def 就是开始定义VGG16的网络

src/utils/callbacks.py

@@ -1,6 +1,6 @@
 import os
 
-import keras
+import tensorflow as tf
 import matplotlib
 matplotlib.use('Agg')
 from matplotlib import pyplot as plt
@@ -8,7 +8,7 @@ import scipy.signal
 import tensorflow as tf
 
 
-class LossHistory(keras.callbacks.Callback):
+class LossHistory(tf.keras.callbacks.Callback):
     def __init__(self, log_dir):
         import datetime
         curr_time = datetime.datetime.now()
@@ -60,7 +60,7 @@ class LossHistory(keras.callbacks.Callback):
         plt.cla()
         plt.close("all")
 
-class ExponentDecayScheduler(keras.callbacks.Callback):
+class ExponentDecayScheduler(tf.keras.callbacks.Callback):
     def __init__(self,
                  decay_rate,
                  verbose=0):

src/utils/dataloader.py

@@ -2,15 +2,15 @@ import math
 from random import shuffle
 
 import cv2
-import keras
+import tensorflow as tf
 import numpy as np
-from keras.utils import to_categorical
+from tf.keras.utils import to_categorical
 from PIL import Image
 
 from .utils import cvtColor, preprocess_input
 
 
-class ClsDatasets(keras.utils.Sequence):
+class ClsDatasets(tf.keras.utils.Sequence):
     def __init__(self, annotation_lines, input_shape, batch_size, num_classes, train, **kwargs):
         super().__init__()
         self.annotation_lines   = annotation_lines