vc++ 调用tensorflow 2.0 的.pb模型进行神经网络预测

vc++(vs2017版本)下调用python生成的tensorflow2.0 xx.pb模型进行神经网络图片预测。

代码:

#pragma once

class CMyAiMode
{
public:
   CMyAiMode();
   ~CMyAiMode();
public:
   void Test();
};

#include "pch.h"
#include "MyAiMode.h"
#include <opencv2/core/core.hpp>
#include <opencv2/opencv.hpp>
#include <opencv2/dnn.hpp>
#include <iostream>
using namespace cv;
using namespace std;
using namespace cv::dnn;

CMyAiMode::CMyAiMode()
{
}

CMyAiMode::~CMyAiMode()
{
}

void CMyAiMode::Test()
{
    std::string strPath;
   strPath = "D:\\5.jpg";
   Mat src_image = imread(strPath);
   if (src_image.empty())
   {
       return;
   }
   size_t num_c = src_image.channels();
   cv::Size sz = src_image.size();
   string path = "D:\\weld.pb";
   Net net = readNetFromTensorflow(path);
   double d = 1.0 / 255;
   cv::Scalar mean{ 0, 0, 0 };
   bool swapRB{ true };//关键
   int inputW = 100;//模型图片宽高
   int inputH = 100;
   Mat blob = blobFromImage(src_image, d, cv::Size(inputW, inputH), mean, swapRB, false);
   net.setInput(blob);
   Mat out = net.forward();
   TRACE(_T("cols:%d %d elemsize:%d channels:%d, total:%d result:%.6f %.8f\n"), out.cols, out.rows, out.elemSize(), out.channels(), out.total(), out.at<float>(0,0), out.at<float>((0,1)));
   float f, f2;//预测结果为两个标签, 各自概率
   memcpy(&f, out.data, sizeof(float));
   memcpy(&f2, out.data+4, sizeof(float));
    TRACE(_T("%.6f %.6f"), f, f2);
}

opencv库下载地址:https://download.csdn.net/download/wcqq12345/87249348

vc++ 调用tensorflow 2.0 的.pb模型进行神经网络预测相关推荐

opencv调用TensorFlow1.x训练的pb模型
文章目录前言一.搭建虚拟环境二.通过frozen_pb来生成pbtxt文件 1.获取opencv官方的转化文件 2.生成pbtxt文件三.OpenCV调用pb模型前言通过opencv读取p ...
第五章：Tensorflow 2.0 利用十三层卷积神经网络实现cifar 100训练（理论+实战）
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明. 本文链接:https://blog.csdn.net/LQ_qing/article/deta ...
c调用python keras模型_使用TensorFlow 2.0创建Keras模型的三种方法
TensorFlow 2.0和tf.keras提供了三种方式来实现神经网络模型:Sequential API Functional API Model subclassing 下面我将分别使用这三种方 ...
VS2015+OpenCV3.4.5+QT5.12+WINDOWS10用c++调用tensorflow训练好的.pb文件图像检测
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明. 本文链接:https://blog.csdn.net/qq_31806049/article/ ...
【TensorFlow系列】【五】利用inception v3 pb模型文件做预测
2019独角兽企业重金招聘Python工程师标准>>> 本文介绍如何利用imagenet比赛上训练好的inception v3冻结的pb模型进行inference. 1.下载ince ...
tensorflow 1.0 学习：模型的保存与恢复(Saver)
将训练好的模型参数保存起来,以便以后进行验证或测试,这是我们经常要做的事情.tf里面提供模型保存的是tf.train.Saver()模块. 模型保存,先要创建一个Saver对象:如 saver=tf. ...
TensorFlow 1.0.0rc1上玩量化神经网络 ——转自慢慢学TensorFlow 微信公众号
注这里的量化不是指"量化交易(Quantitative trade)",而是 Quantization ,即离散化,注意是否走错片场. 前言开发神经网络时最大的挑战是让它真正起 ...
Tensorflow训练神经网络保存*.pb模型及载入*.pb模型
1 神经网络结构 1.0 保存*.pb模型 import tensorflow as tf from tensorflow.python.framework import graph_util fro ...
读取pb模型进行预测
本程序是解析一个tfrecord文件数据,然后调用训练好的pb模型文件去预测这些数据的类别,返回一个列表. 之前的训练程序和需要的数据到可以到这儿找:https://blog.csdn.net/mac ...

vc++ 调用tensorflow 2.0 的.pb模型进行神经网络预测

vc++ 调用tensorflow 2.0 的.pb模型进行神经网络预测相关推荐

最新文章

热门文章