gpt4 book ai didi

python - 将图层添加到 keras 预训练模型中

转载 作者:行者123 更新时间:2023-12-05 08:30:19 27 4
gpt4 key购买 nike

<分区>

我正在尝试构建一个 ResNet50 CNN,用于将图像分类为 5 个不同的类别。我首先导入这个模型:

ResNet = ResNet50(
include_top= None, weights='imagenet', input_tensor=None, input_shape=([128, 217, 3]),
pooling=None, classes=5)

然后我尝试为分类添加一些最终层:

ResNet.add(Flatten())
ResNet.add(Dense(units=512, activation='relu'))
ResNet.add(Dropout(0.5))
ResNet.add(Dense(units=256, activation='relu'))
ResNet.add(Dropout(0.5))
ResNet.add(Dense(units=5, activation='softmax'))

但是当我尝试这个时,我得到一个错误:

AttributeError: 'Functional' object has no attribute 'add'

有人知道怎么解决吗?或者我应该如何尝试添加图层?

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com