是否可以在TensorFlow上加载学习模型(.t7)?

时间:2018-01-13 09:43:30

标签: python python-3.x tensorflow deep-learning torch

我已经学习了CNN的模型(model.t7),我想在另一个基于Python的深度学习框架(例如,TensorFlow)上使用这个模型。

你能告诉我如何实现它吗?

谢谢。

1 个答案:

答案 0 :(得分:2)

那里有一些脚本/库。

您可在此处找到概述:https://github.com/ysh329/deep-learning-model-convertor

在您的情况下,您需要转换:

  1. 从火炬到pyTorch https://github.com/clcarwin/convert_torch_to_pytorch
  2. 从pyTorch到Tensorflow(在Keras上)https://github.com/nerox8664/pytorch2keras
  3. 但事先阅读了这些限制,因为不支持所有操作。