tensorflow C++图像分割/目标检测从tensor输出图像

以前我是由tensorflow C++得到图像分类结果，即得到不同类别的概率，如tensorflow c++ API预测多张图片batch inference_元气少女缘结神的博客-CSDN博客所示，这种方法用在很多地方都没问题，并与python下结果一致，包括小数点后四五位仍旧保持概率一致。

以前对于图像分类的应用中，就是按下面这样从预测后的tensor得到概率：

auto tmap = probabilities.tensor<float, 2>();

但是tensorflow C++中对于图像分割/目标检测这种应用，需要从tensor中输出处理结果即二维图像，以图像分割为例，需要输出经过分割后的图像数据矩阵而非一个概率。这方面资料太少，无法像上面那样写，否则是得不到真正正确的分割后的图像，即使得到了也是错误的或者与python下分割图像不一致。

摸索很久终于搞定，现在与python下图像分割结果已经一致。如下图所示：左边是python输出的分割图像，右边是tensorflow C++输出的分割图像。

可以看到TF C++图像分割结果与python下TF图像分割结果一致。

整体代码放在 https://download.csdn.net/download/wd1603926823/87517527 这个链接中，大家按需查看。我用的是自己编译的tensorflow gpu c++库 ubuntu下tensorflow 2.0/2.5 c++动态库编译gpu版本_元气少女缘结神的博客-CSDN博客_tensorflow2.0 ubuntu 后面测试过用tensorflow cpu c++库也一样 opencv4.1无法加载python-cnn模型，编译第三方库libtensorflow_cc.so巨坑_元气少女缘结神的博客-CSDN博客

无论是CPU库还是GPU库 2.0~2.5都可以，主要是按链接 https://download.csdn.net/download/wd1603926823/86954185 中的代码写就行了。看性能，以前我用的是TF-CPU，绝对是达不到下图的性能的，所以尽量用TF-GPU，别用TF-CPU：

tensorflow C++图像分割/目标检测从tensor输出图像相关推荐

Ubuntu Tensorflow object_detection API 目标检测环境搭建
Ubuntu 16.04下安装TensorFlow Object Detection API(对象检测API) Ubuntu 16.04下搭建TensorFlow运行环境(用Anaconda) ...
OpenCvSharp (C# OpenCV) DNN模块加载自己训练的TensorFlow模型做目标检测（含手势识别、骰子识别、菜品识别）(附源码)
本文作者Color Space,文章未经作者允许禁止转载! 本文将介绍OpenCVSharp DNN模块加载自己训练的TensorFlow模型做目标检测(含手势识别.骰子识别.菜品识别)! 前言: 下 ...
Tensorflow faster RCNN目标检测车牌
主要是利用Tensorflow官方examples提供的API来实现目标检测,官方examples,打开链接你会发现官方其实已经实现了很多的例子了,包括目标检测,自编码器,OCR,GAN,img2tx ...
tensorflow精进之路(二十三)——Object Detection API目标检测(上)（Fast R-CNN算法）
1.概述上一讲,我们使用slim库对图片进行检测,每个物品用同一种颜色标注,显得乱七八糟的.这一讲,我们来学习目标检测.目标检测就是,输入一张图片,输出是将该图片中所含的所有目标物体识别,并标记出他 ...
深度学习/自动驾驶数据集大集合（目标检测/图像分割/语义分割/图像分类/）
CIFAR-10数据集 CIFAR-10数据集(加拿大高级研究所,10类)是 Tiny Images 数据集的一个子集,由60000个32x32彩色图像组成.这些图片被贴上了10个相互排斥类别的标签: ...
深度学习（十四）——Softmax详解, 目标检测, RCNN
https://antkillerfarm.github.io/ Softmax详解首先给出Softmax function的定义: yc=ζ(z)c=ezc∑Cd=1ezd for c=1,-,C ...
目标检测，FFmpeg中第一个基于深度学习模型的视频分析功能
2021年4月,终于把目标检测(object detection)加到FFmpeg upstream了,有maintainer身份加持,还是交互了将近100封邮件,花了两个多月才完成upstream, ...
tensorflow精进之路(二十五)——Object Detection API目标检测(下)（VOC数据集训练自己的模型进行目标检测）
1.概述上一讲,我们使用了别人根据COCO数据集训练好的模型来做目标检测,这一讲,我们就来训练自己的模型. 2.下载数据集为了方便学习,我们先使用别人整理好的数据集来训练---VOC 2012数据 ...
计算机视觉笔记及资料整理(含图像分割、目标检测)
前言 1.简单聊聊: 在我脑海中我能通过这些年听到的技术名词来感受到技术的更新及趋势,这种技术发展有时候我觉得连关注的脚步都赶不上.简单回顾看看,从我能听到的技术名词来感受,最开始耳闻比较多「云计算」 ...

tensorflow C++图像分割/目标检测从tensor输出图像

tensorflow C++图像分割/目标检测从tensor输出图像相关推荐

最新文章

热门文章