노트/Python : 프로그래밍
[케라스 오류 해결] TypeError: The added layer must be an instance of class Layer
Diane_
2020. 4. 24. 15:50
# 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'),
])
이 코드 실행하고 오류 부분 다시 실행 해보기