Keras: How to use fit_generator with multiple inputs

后端 未结 2 573
深忆病人
深忆病人 2020-12-02 21:03

Is it possible to have two fit_generator?

I\'m creating a model with two inputs, The model configuration is shown below.

Label Y uses the same label

2条回答
  •  长情又很酷
    2020-12-02 21:15

    Try this generator:

    def generator_two_img(X1, X2, y, batch_size):
        genX1 = gen.flow(X1, y,  batch_size=batch_size, seed=1)
        genX2 = gen.flow(X2, y, batch_size=batch_size, seed=1)
        while True:
            X1i = genX1.next()
            X2i = genX2.next()
            yield [X1i[0], X2i[0]], X1i[1]
    

    EDIT after Thanh Nguyen comment

    Generator for 3 inputs:

    def generator_three_img(X1, X2, X3, y, batch_size):
        genX1 = gen.flow(X1, y,  batch_size=batch_size, seed=1)
        genX2 = gen.flow(X2, y, batch_size=batch_size, seed=1)
        genX3 = gen.flow(X3, y, batch_size=batch_size, seed=1)
        while True:
            X1i = genX1.next()
            X2i = genX2.next()
            X3i = genX3.next()
            yield [X1i[0], X2i[0], X3i[0]], X1i[1]
    

提交回复
热议问题