Я создаю класс, который включает в себя архитектуру mobil enet из библиотеки Keras.
### THE MOBILENET PIPELINE
from keras.applications.mobilenet import MobileNet
class mobilenet:
def __init__(self,
weights = None,
include_top = False,
input_shape = (960,604,3),
drop_lastlayers = 5,
num_outputnodes = 1000,
reuse=False):
#self.parameters = []
self.weights = weights
self.include_top = include_top
self.input_shape = input_shape
self.drop_lastlayers = drop_lastlayers
self.num_outputnodes = num_outputnodes
self.reuse = reuse
#self.set_parameters()
def forward(self, data):
## INITIALIZATION of THE MOBILENET
mnet = keras.applications.mobilenet.MobileNet()
## CONVERTING IT INTO SEQUENTIAL TYPE OF MODEL
classifier = Sequential()
for layer in mnet.layers:
classifier.add(layer)
## CHANGING THE OUTPUT LAYER
classifier.add(Dense(self.num_outputnodes, activation='softmax'))
## TO KEEP INITIAL WEIGHTS
for layer in classifier.layers:
layer.trainable = False
## MODEL CREATION
inputs = classifier.input
outputs = [classifier.layers[i].output for i in range(len(classifier.layers))]
model = Model(inputs, outputs)
## PREDICTIONS
self.all_layers_predictions = model.predict(data)
def get_layer(self,index):
return self.all_layers_predictions[index] #getting the result of the layer[index]
mnet = mobilenet()
Затем я создаю заполнитель TF, чтобы сохранить изображение для передачи по сети *
Однако я получил следующую ошибку:
ValueError: при использовании тензоров данных в качестве входных данных для модели вы должны указать аргумент steps
.
Как я могу решить эту проблему проблема?