cnn 실습_나만의 이미지 데이터를 만들고 학습하기(3) - functional API & save_model
2020. 3. 26. 14:25
# Functional API를 사용하여 model 구성 def create_model2(): inputs = keras.Input(shape=(128, 128, 3)) conv1 = keras.layers.Conv2D(filters=32, kernel_size=[3, 3], padding='SAME', activation='relu')(inputs) pool1 = keras.layers.MaxPool2D(padding='SAME')(conv1) conv2 = keras.layers.Conv2D(filters=64, kernel_size=[3, 3], padding='SAME', activation='relu')(pool1) pool2 = keras.layers.MaxPool2D(padding='SAME..