1. 구현 방법

1) 데이터 Pipeline

2) 모델 구현

    base_model = tf.keras.applications.MobileNetV2(input_shape=(224, 224, 3),
                                                   include_top=False,
                                                   weights='imagenet')

    # layer의 output tensor
    x = base_model.layers[-1].output
    x = tf.keras.layers.GlobalAveragePooling2D()(x)
    x = tf.keras.layers.Dense(2, activation="softmax")(x)

    model = tf.keras.Model(inputs=base_model.input, outputs=x)
    model.summary()

    model.compile(optimizer=tf.keras.optimizers.RMSprop(lr=0.0001),
                  loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False),
                  metrics=['accuracy'])

3) training loop

2. Trouble Shooting

tf.keras.Model 객체 생성할 때 attributeerror: 'tuple' object has no attribute 'layer'

• 문제

  • keras 와 tensorflow-2.x 가 함께 설치되어있을때 나옴

• 해결책

  • keras를 지우고 tf.keras 를 사용
  • tf.keras 대신 keras를 사용