2017-04-25 60 views
1

我目前生成一個data_generate(batch_size),它需要 batch_size作爲參數。keras fit_generator用於多個批次和多個輸入

我的網絡是多輸入網絡,其具有形狀(45,8,3)

如果批量大小= 1,那麼我可以很容易地得到我的輸入數據作爲一個[33 inputs] 33個輸入,但當批量大小變> 1,是當我不知道如何格式化我的輸入數據。

這樣做[[33 inputs], [33 inputs], [33 inputs] , [33 inputs], .. ,[33 inputs] ]似乎並不奏效,因爲keras期望列表中的第一個條目是numpy.ndarray,在這種情況下,它是numpy ndarrays的列表。

我可以給它的形狀(33,45,8,3)的numpy的陣列的列表,所以像[(33,45,8,3),(33,45,8,3),(33,45,8,3),....,(33,45,8,3)]

但這會導致當我2測試了這個錯誤:

ValueError: Error when checking model input: the list of Numpy arrays that you are passing to your model is not the size the model expected. Expected to see 33 arrays but instead got the following list of 2 arrays: [array([[[[ 3., 4., 4.], 

那麼期望的輸入格式是什麼?

回答

1

相信對於你的輸入所需的格式是:

形狀(batch_size, 45, 8, 3) 33個numpy的陣列的列表中,第一尺寸將是批量。