首页 文章

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

提问于
浏览
2

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

有可能或我该怎么办?

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

谢谢

1 回答

  • 1

    如果您的模型完全是CNN,则完全不需要具有不同的模型 .

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

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

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

相关问题