Как убрать последние слои модели Inception - PullRequest
1 голос
/ 08 марта 2020

Привет! Я следил за форумом от Github об удалении последних слоев предварительно обученной модели. Однако это не работает для меня; возможно, я сделал что-то не так

Я слежу за этим и вот мой код. Я думал, что все, что я должен был сделать, это model.layers[-2].output, но это говорит мне AttributeError: 'Tensor' object has no attribute 'summary'

import tensorflow as tf
import tensorflow.keras as keras
import tensorflow.keras.backend as K
import numpy as np
from tensorflow.keras.layers import Dense, Input, Layer
from tensorflow.keras.models import Model
from tensorflow.keras.applications.inception_v3 import InceptionV3


model = InceptionV3()
print(model.summary())

modele = model.layers[-2].output
print(modele.summary())

1 Ответ

1 голос
/ 08 марта 2020

переменная модели - это только слой. Вы должны сделать:

model = InceptionV3()
print(model.summary())

output = model.layers[-2].output
modele = Model(inputs = model.input, outputs = output)
print(modele.summary())
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...