风格迁移:在 tensorflow 1.15.0 中保存和恢复检查点/模型

Posted

技术标签:

【中文标题】风格迁移:在 tensorflow 1.15.0 中保存和恢复检查点/模型【英文标题】:Style Transfer : Save&Restore checkpoint/model in tensorflow 1.15.0 【发布时间】:2020-03-09 19:48:20 【问题描述】:

我对在 tensorflow 1.15.0 中保存和恢复模型感到有些沮丧。我想在 jupyter notebook / google colab notebook 环境中实现它。该应用程序是图像的样式转换。

我只是想保存模型并恢复它,以便为更多图像应用样式转换。

tensorflow 文档有点混乱,(我没有找到这方面的例子),所以我从来不知道正确的语法是什么样的。

我现在想实现一件事:

    恢复模型 正确。

我现在将写相关的行:

model = get_model()  
opt = tf.train.AdamOptimizer(learning_rate=2.5,beta1=0.99, epsilon=1e-1)
saver = tf.train.Checkpoint(model=model, optimizer=opt)
saver.save('/content/sample_data/test/_____NEU____')

当我想恢复模型时,我使用命令:

saver.restore('/content/sample_data/test/_____NEU____')

如何解决此问题并正确加载我的检查点文件?谢谢


google colab 项目在这里:

https://colab.research.google.com/drive/12hTitoQ2-tH8pYEsfMDR5jtsg8a96PgC


【问题讨论】:

微信聊天:pastebin.com/nb6mPcFa 【参考方案1】:

我还尝试通过 keras - 语法保存模型。

model.load_weights('/content/sample_data/saved_model/my_model6.h5')

model.save_weights('/content/sample_data/saved_model/my_model6.h5', save_format='h5')

没有错误消息,但模型没有加载之前训练的权重。昨天我在 mirc-channel 中询问了某人,他说这可能是 tensorflow 1.15.0 的 tf.keras 中的一个错误,但我昨天在 tensorflow 1.14.0 和 1.13.1 上进行了尝试,但没有成功。

【讨论】:

【参考方案2】:

这种技术不会改变模型,但您可以更新输入图像以产生所需的输出。 ..所以保存没有问题,但是模型没有改变。

【讨论】:

以上是关于风格迁移:在 tensorflow 1.15.0 中保存和恢复检查点/模型的主要内容,如果未能解决你的问题,请参考以下文章

TensorFlow实时任意风格迁移,送女朋友的创意礼物有了

照片瞬间变样!看Tensorflow如何实现深度「图片风格迁移」

tensorflow_hub实现风格迁移

Tensorflow2.5—入门实践综合篇——物体识别花朵分类猫狗识别风格迁移文本分类

风格迁移

OpenCV DNN 模块-风格迁移