caffe 转tensorflow padding 的区别
博主在对tensorflow转caffe的研究中遇到了很多问题,一直想做一个自动化的脚本,能直接从tensorflow 转出caffe 的prototxt和caffemodel,目前进展较为顺利,特地写一系列的博客,来帮助后来者,可能很快就不用我们亲自去转了,因为现在都在提倡通用话,后期都会转到onnx上面去.
如果pad mode 为valid的话就可以忽略了,都是一样的.
大家在转tensorflow到caffe 的过程中要注意到,caffe 的padding和tensorflow的padding 的方式有所不同.尤其是在当input size 为偶数时,最后的tensorflow 可能只需要在右侧或者下测添加一行或者一列全是0的数.而caffe 则会在左侧护着上侧添加一行或者一列0.这一个区别会影响后续的转化,所以我们需要对此处理,处理方案为
- 修改tensorflow 源码,不靠谱,这样你的ckpt就要重新训练
- 修改caffe 源码,无法做到通用性.
- 还在想...一旦想起来就立刻更新....
由于这是和公司合作的脚本,所以暂时不能放送,有问题可以联系博主:lucky_lsq@163.com