用Keras有效地逐一收集小批量

时间:2017-06-06 09:22:02

标签: tensorflow keras mini-batch

我必须建立一个接收不同大小图像的网络。我不想调整大小或裁剪,所以我使用完全卷积网络。

问题在于,由于每张图片的大小不同,我无法预先创建小批量广告。

一种解决方案是在预期的小批量中拍摄最大的图像,并将所有其他图像填零以适合相同的尺寸。然而,这在时间和记忆方面效率不高,特别是因为图像尺寸明显不同(甚至30px到3000px)。

我现在使用的另一个解决方案是创建1的小批量,这当然解决了不同大小的问题,但它对收敛没有好处。

所以问题是Keras是否提供了一些从几个输入中收集渐变的方法,然后才采取学习步骤?

0 个答案:

没有答案