在OpenCV中使用Nvidia DIGITS训练模型时出错

时间:2018-04-24 09:37:19

标签: c++ opencv caffe nvidia-digits

我使用DIGITS(NVCaffe)训练我的模型,我在opencv中使用它,类似于使用.prototxt&的opencv示例。 .caffemodel和我测试它并且工作得很好。 但是当我使用由数字训练的模型时,我得到了这个错误:

OpenCV Error: Assertion failed (pbBlob.raw_data_type() == 
caffe::FLOAT16) in blobFromProto, file 
/opt/opencv/modules/dnn/src/caffe/caffe_importer.cpp, line 242
Exception:  /opt/opencv/modules/dnn/src/caffe/caffe_importer.cpp:242: 
error: (-215) pbBlob.raw_data_type() == caffe::FLOAT16 in function 
blobFromProto

帮我解决一下! 很多次提到这个问题没有回答! http://answers.opencv.org/question/177086/unable-to-use-caffe-model-trained-in-nvidia-digits-in-opencv-dnn-code/

1 个答案:

答案 0 :(得分:0)

OpenCV支持以BVLC格式而非NVCaffe训练的模型。尝试通过将DIGITS中的求解器参数从 NVCaffe 更改为 Compatible 来训练模型。