Caffe2 C++ 预测Demo

最新推荐文章于 2020-02-25 21:00:13 发布

Joson_yuan

最新推荐文章于 2020-02-25 21:00:13 发布

阅读量2.2k

点赞数

本文链接：https://blog.csdn.net/Joson_yuan/article/details/79380130

版权

本文详细介绍了如何使用Caffe2的C++接口进行模型预测，通过实例代码展示了预测过程，帮助读者理解Caffe2在实际应用中的操作步骤。

摘要由CSDN通过智能技术生成

Caffe2预测代码：

bool FingerprintDNNLocating::initCaffe2(void)
{
	string mdPath = map_mgr_.getCaff2ModelPath();
	if ("" == mdPath)
		return false;
	string initNetPath = mdPath + "/init_net.pb";
	string predictNetPath = mdPath + "/predict_net.pb";
	if (!std::ifstream(initNetPath).good() ||
		!std::ifstream(predictNetPath).good()) 
	{
		return false;
	}
	CAFFE_ENFORCE(ReadProtoFromFile(initNetPath, &initNet_));
	// >>> with open(path_to_PREDICT_NET) as f:
	CAFFE_ENFORCE(ReadProtoFromFile(predictNetPath, &predictNet_));
    predictor_ = new caffe2::Predictor(initNet_, predictNet_);
	return true;
}

bool FingerprintDNNLocating::runCaffe2(vector<double> &weights)
{
    if(!predictor_)
        return false;
    caffe2::TensorCP