我想训练CNN,其中图像尺寸为128 * 512,然后我想使用此权重文件来训练其他具有128 * 1024尺寸的数据。这意味着我想在不同数据(128 * 1024)的训练时间内使用预先训练的重量文件。
有可能或我该怎么做?
我想这样做是因为我只有300张尺寸为128 * 1024的图像,而我有5000幅图像,尺寸为128 * 512,两个数据集都不同。
谢谢
答案 0 :(得分:1)
如果你的模型完全是CNN,那么绝对不需要有不同的模型。
CNN模型可以拍摄任何尺寸的图像。只需确保input_shape=(None,None,channels)
你需要单独的numpy数组,一个用于大图像,另一个用于小图像,你必须为每个数组调用不同的拟合方法。
(但可能你会使用发电机)