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

Ken*_*Ken 2 python python-3.x deep-learning torch tensorflow

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

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

谢谢。

sla*_*mic 5

那里有一些脚本/库。

您可以在这里找到概述:https ://github.com/ysh329/deep-learning-model-convertor

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

  1. 从 Torch 到 pyTorch https://github.com/clcarwin/convert_torch_to_pytorch
  2. 从 pyTorch 到 Tensorflow(通过 Keras)https://github.com/nerox8664/pytorch2keras

但请事先阅读限制,因为并非所有操作都受支持。