重量文件用于不同尺寸的图像

时间:2018-05-17 13:04:27

标签: python keras deep-learning convolutional-neural-network

我想训练CNN,其中图像尺寸为128 * 512,然后我想使用此权重文件来训练其他具有128 * 1024尺寸的数据。这意味着我想在不同数据(128 * 1024)的训练时间内使用预先训练的重量文件。

有可能或我该怎么做?

我想这样做是因为我只有300张尺寸为128 * 1024的图像,而我有5000幅图像,尺寸为128 * 512,两个数据集都不同。

谢谢

1 个答案:

答案 0 :(得分:1)

如果你的模型完全是CNN,那么绝对不需要有不同的模型。

CNN模型可以拍摄任何尺寸的图像。只需确保input_shape=(None,None,channels)

即可

你需要单独的numpy数组,一个用于大图像,另一个用于小图像,你必须为每个数组调用不同的拟合方法。

(但可能你会使用发电机)