预训练模型进行微调
定义模型
class MyModel(tf.keras.Model):
def __init__(self, height=None,width=None,channel=None):
super(MyModel,self).__init__()
self.inputshape = tf.keras.layers.InputLayer(input_shape=(height,width,channel))
# self.resnet50 = tf.keras.applications.ResNet50(include_top=False,
# weights="imagenet",
# input_shape=(height,width,channel),
# )
#
#
self.effnet = tfkeras.EfficientNetB0(
include_top=False,
weights='imagenet',
input_shape=(height,width,channel)
)
self.pool = tf.keras.layers.GlobalAveragePooling2D(name="GlobalAP")
self.dense3 = tf.keras.layers.De