Windows下加载CAFFE模型的问题

本文分享了作者在尝试将Linux环境下训练的Caffe模型移植到Windows平台时遇到的问题及解决过程。主要问题在于模型加载不完整,通过修改文件打开方式加入了O_BINARY选项后,成功解决了该问题。
摘要由CSDN通过智能技术生成

这段时间试着把linux训练的caffe 模型在windows环境下运行起来,因为是自己改过的caffe代码,又只有VS2012,并没有下载官网的windows版本(支持VS2013及以上)。把相应的caffe源码拷过来,配置好各种库(hdf5/protobuf/boost/glog等),使用protoc.exe生成相应的caffe.pb.h和caffe.pb.cc,使用windows的库函数替换了一些函数,很快就编译成功了。

但在测试过程中,碰到以下问题:

这里写图片描述

错误信息来自:src/caffe/net.cpp中的CopyTrainedLayersFrom(const NetParameter& param)函数,表明加载模型出现了问题。

1、模型会不会与训练代码不匹配?

这个模型是用同样的caffe代码在linux环境下训练的,在linux下测试也没有问题,排除了模型不匹配的问题。

2、代码在windows和linux下有什么不同?

caffe.pb.h和caffe.pb.cc是在windows下重新生成的,差别只在这里了。怀疑protobuf的版本不一致导致的问题,都换为2.5.0版本后,问题仍然存在。
还有一些函数,open/close等,还有一些被注释了的变量,如果有问题,只可能出现在这里了。

* 解决方法 *

在调试过程中,试着加载了几个不同的模型,发现函数中的CopyTrainedLayersFrom中param.layer_size() 都非常小,跟模型的网络层差距很大,问题的原因:模型没有加载完全,导致后续的解析错误。

参照windows下用caffe加载二进制模型(linux下训练)的问题,很快就找到了问题所在,src/util/io.cpp函数中:

int fd = open(filename, O_RDONLY);

open函数使用O_RDONLY在linux下加载二进制模型没有问题,但在windows下加载二进制模型需要加入O_BINARY选项:

int fd = open(filename, O_RDONLY|O_BINARY);

重新编译后,运行成功。

其实如果采用官方的caffe-windows是肯定不会出现上述问题的,无奈自己挖的坑含着泪也要填完。不过在这次纠错过程中,仔细看了之前从来没有注意的caffe源码中的一些细节,也算是因祸得福。

纠错也要有步骤、有逻辑地纠。刚开始眉毛胡子一把抓,比较耽误时间,就差重新下载新 caffe再重新训练模型了。文中参考的链接博客,解决问题时有理有据,值得学习。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值