keras处理任意大小输入

Author Avatar
Patrick 4月 11, 2020

让 keras 处理任意大小输入其实很简单,只需:

1
2
3
4
# for rgb
main_input = layers.Input(shape=(None, None, 3))
# for gray
main_input = layers.Input(shape=(None, None, 1))

对于 U-Net 这种又有下采样又有上采样的就要注意一下了,下采样再上采样得到的结果可能与原来的不同,比如 25 下采样是 12,再上采样就是 24,这样往往会出现一些问题,因此最好让输入经过下采样时不会出现除不尽的情况。