【问题标题】:can't load digits trained caffe model with opencv readnetfromcaffe无法使用 opencv readnetfromcaffe 加载数字训练的 caffe 模型
【发布时间】:2018-10-06 22:38:06
【问题描述】:

我最近从this tutorial 构建了数字,一切正常,我终于训练了我的 AlexNet 模型(还训练了一个 SqueezNet,以便我可以在这里上传模型)!问题是当我从 Digits 下载我的模型时,我无法将其加载到我的程序中进行测试!我已经使用下载的 GoogleNet from this link 测试了我的程序,它工作正常! 我在这个函数中使用 OpenCV readNetFromCaffe 来加载 Caffe 模型

void deepNetwork::loadModel( cv::String  model ,cv::String  weight ,string lablesPath,int ps){
patchSize=ps;
labeslPath=lablesPath;
 try
{
    net = dnn::readNetFromCaffe(weight,model);
    cerr<<"loaded succ"<<endl;
}
catch (cv::Exception& e)
{
    std::cerr << "Exception: " << e.what() << std::endl;

}}

我在加载我的模型时收到以下错误

OpenCV 错误:断言失败 (pbBlob.raw_data_type() == caffe::FLOAT16) 在 blo
bFromProto,文件 /home/nvidia/build-opencv/opencv/modules/dnn/src/caffe/caffe_im
porter.cpp,第 242 行异常: /home/nvidia/build-opencv/opencv/modules/dnn/src/caffe/caffe_importer .cpp:242: 错误: (-215) pbBlob.raw_data_type() == caffe::FLOAT16 in 功能块
bFromProto

OpenCV 错误:未找到请求的对象(请求的 blob“数据” 未找到)我
n 设置输入,文件 /home/nvidia/build-opencv/opencv/modules/dnn/src/dnn.cpp,行
1606 在抛出 'cv::Exception' 实例后调用终止
什么(): /home/nvidia/build-opencv/opencv/modules/dnn/src/dnn.cpp:1606:错误: (-204) 在函数 setInput 中找不到请求的 blob“数据”

中止(核心转储)

任何帮助将不胜感激

opencv 3.3.1 版也在 (3.3.0 ,3.4.1) 上测试过同样的错误! 在没有 Cuda、Cudnn 或 Caffe 的系统上进行测试,只需纯 c++ 和 OpenCv ... 但我已经用 Cuda、Cudnn 和 caffe 在 aws ec2 实例(p3.2xlarge)上训练了我的模型!

you can download the trained squeezNet model (.prototxt and .caffemodel) here

【问题讨论】:

标签: c++ caffe opencv3.1 nvidia-digits


【解决方案1】:

OpenCV DNN 模型需要 BVLC 格式的 caffemodel。但是,NVCaffe 以比 BVLC Caffe 更有效的格式存储 caffe 模型。 如果您想要与 BVLC/Caffe 和 NVcaffe 兼容的模型。

在solver.prototxt中添加这个标志 store_blobs_in_old_format = true

请阅读 DIGITS NVCaffe 文档。 NVCaffe Documenation - store_blobs_in_old_format

【讨论】:

    【解决方案2】:

    终于,我发现了问题! 这是一个版本问题,我有数字 6.1.1 与 nvcaffe 0.17.0 一起用于训练,这与以前的 Caffe 和 OpenCv 库不兼容!你必须将 NvCaffe 降级到 0.15.14 版本,它会很容易用 OpenCv 打开!

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2015-11-28
      • 2015-12-24
      • 1970-01-01
      • 2018-08-28
      相关资源
      最近更新 更多