tensorflow中使用预训练模型进行微调

预训练模型进行微调

定义模型

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