opencv打开pb_opencv调用tensorflow模型(单个pb文件)

本文介绍了如何使用OpenCV直接读取TensorFlow的.pb模型文件,进行图像预处理,输入模型并进行预测,以及解析预测结果。在部署模型时,强调了模型压缩的重要性,同时提供了关键的代码示例。
摘要由CSDN通过智能技术生成

opencv是图像处理的开源库,使用非常方便,所以一般在做图像处理时基本都是会使用的。tensorflow也有基于c++的接口,但是使用时需要进行编译tensorflowc++源码,编译起来比较繁琐。所以一般也不怎么使用。当然如果opencv可以直接调用tensorflow模型那是在好不过的。

一,训练模型的保存

这里需要说一下,模型一般都是比较大的,在部署时需要对模型进行压缩,使程序运行可以更可能的快。模型的压缩可以参考模型压缩,在压缩模型直接保存模型时一定要将图和参数一起保存,否则压缩会有问题的。

二,opencv读取模型

读取模型的主要代码如下

Net net = readNetFromTensorflow(tf_pb_file);

三,图像的预处理

主要代码如下:

Mat inputBlob = blobFromImage(src, 1.0/255. , Size(w, h), Scalar(), true, false);

其原函数如下:

blobFromImage(InputArray image,

double scalefactor=1.0,

const Size& size = Size(),

const Scalar& mean = Scalar(),

bool swapRB = false,

bool crop = false,

int ddepth = CV_32F)

image:将要输入神经网络的图片,未进行处理的图片

mean:需要将图

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值