前几天Google放出了pretrained好的Inception ResNet v2模型,在imagenet上面效果非常赞,现在我想把这个TF模型转成caffe。

没有玩过TF,所以比较困惑。如果要自己对照TF里面的param name来写prototxt的话感觉要歇菜(1w多行的prototxt)~不知道有没有什么高效的工具? 求助各位大神~多谢


微软就做了一套能在不同的深度学习框架之间进行互动式操作的工具集——MMdnn,它是一个综合性的跨框架解决方案,能够用于深度神经网路模型的转换,可视化及诊断等操作,可在Caffe,Keras,MXNet,Tensorflow,CNTK和PyTorch等框架之间实现模型的转换。

这里是它的GitHub链接:Microsoft/MMdnn

能在不同的深度学习框架之间转换模型?微软的MMdnn做到了 - 云+社区 - 腾讯云?

cloud.tencent.com


楼主解决了吗?我也想用tensorflow的model在caffe上使用


https://zhuanlan.zhihu.com/p/115081459 可以尝试下?????


同求,貌似比较困难


caffe 转tensorflow相对简单点,逆过程就比较困难了,求高人指导


踩个沙发

前段时间摸索著转过一次,半手工半自动的转,最终转成功了,转的是2019年cvpr的pfld的tensorflow模型 https://blog.csdn.net/pc9803/article/details/100055213

ps:目前,mmdnn好像还不太行,有些tf的层还转不了,我尝试的时候bn层就报各种问题


楼主解决了么?


推荐阅读:
相关文章