Ошибка значения при загрузке тензора в модель Keras Sequential - PullRequest
0 голосов
/ 09 июля 2020

Я создаю класс, который включает в себя архитектуру 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.

Как я могу решить эту проблему проблема?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...