[케라스 오류 해결] TypeError: The added layer must be an instance of class Layer

2020. 4. 24. 15:50노트/Python : 프로그래밍

 

# TypeError: The added layer must be an instance of class Layer

from keras.layers import *
from keras.models import Sequential
from keras.applications.resnet50 import ResNet50

CLASS_COUNT = 10

base_model = ResNet50(
    weights='imagenet',
    include_top=False, 
    input_shape=(224, 224, 3), 
    pooling='avg',
)
base_model.trainable = False

model = Sequential([
  base_model,
  Dense(CLASS_COUNT, activation='softmax'),
])

이 코드 실행하고 오류 부분 다시 실행 해보기