Создание собственного генератора, дающего два входа в NN - PullRequest
0 голосов
/ 06 января 2020

Мне нужно создать собственный генератор, который бы вводил текстовый + графический ввод для моей модели (VQA - Визуальный ответ на вопрос для набора данных CLEVR). Выходом является категория («1», «да» и т. Д. c.). И я не знаю, как подать генератор в сеть - он говорит мне «неправильная форма входа» или «неправильный вход»

def gen():
    while True:
        n = random.randint(0,len(labels['questions'])-1)
        def get_image(n):
            img = image_preprocessing(cv2.imread(img_path + '/' + labels['questions'][n]['image_filename']),random.random())
            return img

        def get_question(n):
            _question = text_preprocessing(labels['questions'][n]['question'])
            return _question

        def get_label(n):
            y = labels['questions'][n]['answer']
            return y

        yield [get_image(n), get_question(n)], get_label(n)

Это код генератора. И он не принимает даже данные массива (метод подбора кераса). Итак, вопрос: как подать два входа в модель Tensorflow?

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