tensorflow - VGG19 .h5 文件修改
问题描述
我在修改后的神经传输代码（Gatys 算法）中使用预训练的 VGG19，但我的 PC 不允许我使用原始大小的输入图像（原始高度为 2499 pix，但使用 20GB RAM，我只能使用 1000 pix最大）
在我阅读时，我的解决方案是减少 batch_size。所以，我的问题是 - 如何修改 VGG19 .h5 文件以更改其中的 batch_size？或者也许我可以在我的代码中覆盖它的 batch_size ？
解决方案
假设预训练模型定义在 ImageNet 上，单个样本的最大输入数据大小为 224*224。
如果您尝试传递大量输入，您的深度学习框架可能会将其重塑为许多图像以一次进行分类。
将输入数据的大小调整为 224*224，您将使用单个图像（批量大小为 1）运行。
您可以对模型进行自定义实现以采用更大的输入大小。然而，根据任务，缩小到 224*224 通常会得到很好的结果。