2019年8月23日 下午2:40
Tensorflow转Caffe有什么比较好的方法吗?_算子与模型_Ascend_产品交流_华为云
- 想请教一下各位大神,有没有Tensorflow转Caffe模型的比较方便的方法?
- 我这里有一个Tensorflow卷积网络模型想转Caffe(网络是自己搭的,不是常见的结构)。
- Github上是有些项目,但是比较简单的网络结构,像VGG,只有卷积+池化等,如果要转化一个更复杂网络(更多layer,更复杂拓扑),工作量还是有点大
- 还有一个工具是微软的mmdnn,但是本人踩坑后觉得它还是比较局限,首先环境需要是Tensorflow 1.8及以上,并且并不是所有的网络结构都能转换,比如tf.image.resize_bilinear,所以只好先替换成别的支持的层,跑通,然后手动修改工具生成的模型文件。