deeplabv3 + mobilenetv2 做语义分割并封装成c++部署到移动端，linux，windows等平台(史上最详细)

训练

Deeplab项目安装以及测试

首先为了确保版本支持，先得确认你的tensorflow的版本是1.10以上的。我的linux系统上装的是1.14的tensorflow，因为我一直用的这个版本。

克隆deeplab项目

git clone https://gitee.com/yujiahao123/models2

这是我为了速度fork的deeplab，因为tensorflow的github clone实在太慢了，老是中断，特别难受，当开始我还不知道有码云这种神器，让同事Roland兄弟帮我从加拿大下载然后发邮件给我才拿到，哈哈。

添加项目依赖路径

sudo gedit ~/.bashr

在这个文件最后加上一句

export
PYTHONPATH=/home/william/models/research/slim:/home/william/models/research:$PYTHONPATH

这边把路径改成你自己的
然后激活一下环境

source ~/.bashrc

测试deeplab

cd /home/william/models/research/deeplab

执行

python model_test.py

如果最后输出了OK就说明安装成功了

数据集处理

我的数据是同事Roland兄弟给我的(顺便一提加拿大到中国的文件传输实在太恶心了,下载几个小时百分之八十的时候下载失败伤不起啊)，但是要做处理才能使用，于是我写了一些代码来处理数据，来保证和官方提供的pascal voc格式一致

将数据转换成TFRecord

创建 tfrecord

mkdir tfrecord

将上述制作的数据集打包成TFRecord，使用的是build_voc2012_data.py 在目录/home/bai/models/research/deeplab/datasets下执行

python build_voc2012_data.py
--image_folder="/home/william/dataset/yoho/Images" --semantic_segmentation_folder="/home/william/dataset/yoho/masks" --list_folder="/home/william/dataset/yoho/index"
--image_format="jpg"
--output_dir="/home/william/dataset/yoho/tfrecord"

image_folder ：数据集中原输入数据的文件目录地址
semantic_segmentation_folder：数据集中标签的文件目录地址
list_folder : 将数据集分类成训练集、验证集等的指引目录文件目录
image_format : 输入图片数据的格式，CamVid的是png格式
output_dir：制作的TFRecord存放的目录地址(自己创建)

网络训练

在datasets/data_generator.py文件中，添加camvid数据集描述：

_YOHO_INFORMATION = DatasetDescriptor(splits_to_sizes={'train': 256,  # num of samples in images/training'val': 49,  # num of samples in images/validation},num_classes=3,ignore_label=255,
)

因为yoho共有3个classes

注册数据集

同时在datasets/data_generator.py文件，添加对应数据集的名称：

_DATASETS_INFORMATION = {    'cityscapes': _CITYSCAPES_INFORMATION,    'pascal_voc_seg': _PASCAL_VOC_SEG_INFORMATION,    'ade20k': _ADE20K_INFORMATION,'yoho':_YOHO_INFORMATION, #自己的数据集 }

修改代码

因为是在DeepLab的基础上ﬁne-tune我们自己的数据集，所以需要修改一些代码

修改train.py

其中有一些选项：

使用预训练的所有权重，设置initialize_last_layer=True
只使用网络的backbone，设置initialize_last_layer=False和 last_layers_contain_logits_only=False
使用所有的预训练权重，除了logits以外。因为如果是自己的数据集，对应的classes不同(这个我们前面已经设置不加载logits), 可设置initialize_last_layer=False和 last_layers_contain_logits_only=True

修改train_utils.py

对应的utils/train_utils.py中，将关于 exclude_list 的设置修改，作用是在使用预训练权重时候，不加载该 logit 层：

exclude_list = ['global_step','logits']
if not initialize_last_layer:exclude_list.extend(last_layers)

下载预训练权重

因为我们的数据比较少，除了数据增强之外，使用别人训练好的模型做fineturn是一个好的选择
在model_zoo上下载预训练模型：
这里因为我打算将模型部署到移动端，所以我选择了专门给移动端设计的mobilenetv2
下载地址：https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/model_zoo.md
我用的是deeplabv3_mnv2_cityscapes_train这个预训练模型

训练

我们现在基本需要的都准备好了，那就可以开始炼丹啦！
训练我们需要运行deeplab下的train.py这个脚本

python train.py
--logtostderr
--train_logdir=/home/william/model/models-master/models-master/research/deeplab/exp/yoho/train3
--dataset_dir=/home/william/dataset/yoho/tfrecord
--training_number_of_steps=100
--train_split="train"
--model_variant="mobilenet_v2"
--output_stride=16
--base_learning_rate=3e-5
--train_crop_size=513,513
--train_batch_size=2
--dataset="yoho"
--tf_initial_checkpoint=/home/william/model/models-master/models-master/research/deeplab/exp/yoho/train2/model.ckpt-520

测试结果可视化

训练我们需要运行deeplab下的vis.py这个脚本

python vis.py
--vis_split="val"
--model_variant="mobilenet_v2"
--vis_crop_size=3000,2000
--output_stride=16
--checkpoint_dir="/home/william/model/models-master/models-master/research/deeplab/exp/yoho/train3"
--dataset="yoho"
--colormap_type="pascal"
--vis_logdir="/home/william/model/models-master/models-master/research/deeplab/exp/yoho/vis"
--dataset_dir="/home/william/dataset/yoho/tfrecord"

跑完了之后我们再/home/william/model/models-master/models-master/research/deeplab/exp/yoho/vis这个目录下就能看一下我们的模型的效果

大致看一下效果还可以

性能评估

评估我们需要运行deeplab下的vis.py这个脚本

python vis.py
--eval_split="val"
--model_variant="mobilenet_v2"
--eval_crop_size=3000,2000
--output_stride=16
--dataset="yoho"
--checkpoint_dir="/home/william/model/models-master/models-master/research/deeplab/exp/yoho/train3"
--eval_logdir="/home/william/model/models-master/models-master/research/deeplab/exp/yoho/eval"
--dataset_dir="/home/william/dataset/yoho/tfrecord"
--max_number_of_evaluations=1

最后会输出一个mloU值，这个值越高表示效果越好，我跑了一下mloU值能达到85左右还行

模型输出

模型输出我们需要运行deeplab下的export_model.py这个脚本

python export_model.py
--logtostderr
--checkpoint_path="/home/william/model/models-master/models-master/research/deeplab/exp/yoho/train3/model.ckpt-100"
--export_path="/home/william/model/models-master/models-master/research/deeplab/exp/yoho/save/frozen_inference_graph.pb"
--model_variant="mobilenet_v2"
--num_classes=3

这个脚本会把tensor的变量转成常量，生成传说中的冻结图，就是frozen_inference_graph.pb这玩意

部署

我们有了frozen_inference_graph.pb模型文件之后我们就可以将我们的语义分割网络部署在各个平台了，接下来我会从安卓，linux，windos等各个平台来说怎么把一个论文提出的东西变成一个实实在在的产品

移动端

模型转换

移动端tensorflow的官网提供了适合手机的轻量封装库tensorflow lite，要使用这个tensorflow lite必须先将模型从.pb转换为.tfrecord,对于模型转换，tensorflow官网提供了一个工具。

模型转换的史诗级大坑

转换模型的时候有一个大坑，让我踩了快两礼拜，伤不起啊。
刚开始我模型是这样直接转换的

tflite_convert \--output_file=test.lite \--graph_def_file=frozen_inference_graph.pb \--input_arrays=ImageTensor \--output_arrays=SemanticPredictions \--input_shapes=1,3000,2000,3 \--inference_input_type=QUANTIZED_UINT8 \--inference_type=FLOAT \--mean_values=128 \--std_dev_values=128

先不说转换模型能不能转换成功，就算转换成功了，真正部署到各种平台的时候会报错
比如安卓端

这个问题卡了我半天，问同事，问主管，问tensorflow作者都没有解决我的问题
这个是我提的issue
https://github.com/tensorflow/tensorflow/issues/42622
找了很长时间都没有相关的资料
这边卡了很久
靠人不如靠自己
最后解决方法无意间看到了这个issue才解决了我的问题
https://github.com/tensorflow/tensorflow/issues/23747
主要原因是因为tensorflow的前操作和后处理tensorflow lite并不支持
所以如果直接转换模型的话会出现奇奇怪怪的错误
这个issue上是这么转换，sub_2是第二层而ResizeBilinear_2是倒数第二层，跳过了前处理和后处理
然后自己实现了前处理uint转float32和后处理argmax函数，问题解决

tflite_convert \--output_file=./deeplabv3_513.tflite \--graph_def_file=frozen_inference_graph.pb \--input_arrays=sub_2 \--output_arrays=ResizeBilinear_2 \--input_shapes=1,3000,2000,3 \--inference_type=FLOAT

最终转换方式

最后写一下我的转换方式,我是先给输入输出添加了一些signature，然后存成了SaveModel格式，其实和这个冻结图是一样的，不过带上了一些输入输出的信息

def export_saved_model(sess, input, output):output_path = 'seg/'print('Exporting trained model to ', output_path)builder = tf.saved_model.builder.SavedModelBuilder(output_path)input_tensor_info = tf.saved_model.utils.build_tensor_info(input)tensor_info_input = {'images': input_tensor_info}output_tensor_info = tf.saved_model.utils.build_tensor_info(output)tensor_info_outputs = {'output': output_tensor_info}preditction_signature = (tf.saved_model.signature_def_utils.build_signature_def(inputs=tensor_info_input,outputs=tensor_info_outputs,method_name=tf.saved_model.signature_constants.PREDICT_METHOD_NAME))legcy_init_op = tf.group(tf.tables_initializer(), name='legacy_init_op')builder.add_meta_graph_and_variables(sess, [tf.saved_model.tag_constants.SERVING],signature_def_map={'predict_images': preditction_signature})builder.save()print('Successfully export model to %s' % output_path)

然后再转成tensroflow lite可以使用的tfcord模型

tflite_convert \--output_file=./deeplabv3_513.tflite \--saved_model_dir==seg \--input_arrays=sub_2 \--output_arrays=ResizeBilinear_2 \--input_shapes=1,3000,2000,3 \--inference_type=FLOAT

量化(可选)

转成SaveModel之后转成tfcord模型之前可以做量化，可以使你的模型更小，跑的速度更快
量化代码

import tensorflow as tfconverter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
converter.optimizations = [tf.lite.Optimize.OPTIMIZE_FOR_SIZE]
tflite_quant_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_quantized_model)

编译tensorflow lite

首先我们得先把tensorflow的项目下载下来，为了速度我们还是先把项目fork到码云上

git clone https://gitee.com/yujiahao123/tensorflow

这样就能较快的拿到tensorflow的源码
然后先下载tensorflow lite的依赖
这里有一些依赖是国外的资源所以可能得挂VPN有可能下载不了
运行tensorflow/tensorflow/lite/tools/make路径下的download_dependencies.sh脚本，这个脚本会把lite所需要的依赖下载到make下的download文件夹里
然后我们再下载bazel，这是一个类似于CMake一样的编译工具，我们需要这个东西来编译我们的tensorflow lite的库
我们运行下面这行脚本可以得到tensorflow lite的.so动态库
32bit armeabi-v7a:

bazel build -c opt --config=android_arm //tensorflow/lite:libtensorflowlite.so

64bit arm64-v8a:

bazel build -c opt --config=android_arm64 //tensorflow/lite:libtensorflowlite.so

有了这个动态库我们就可以快乐的使用他的接口啦！
我们需要将他的头文件添加进来，下面是官方的说明
Currently, there is no straightforward way to extract all header files needed,
so you must include all header files in tensorflow/lite/ from the TensorFlow
repository. Additionally, you will need header files from
FlatBuffers and
Abseil.
由此可见，我们需要tensorflow\lite还有FlatBuffers和Abseil的头文件
也就是我们之前download的目录下的downloads\flatbuffers\include和downloads\absl这两个目录

安卓端c++代码

#include <jni.h>
#include <string>
#include<cstring>
#include <android/log.h>
#include "tensorflow/lite/model.h"
#include "tensorflow/lite/kernels/register.h"
#include "opencv2/opencv.hpp"
using namespace std;
using namespace cv;string jstringTostring(JNIEnv* env, jstring jstr) {char *rtn = NULL;jclass clsstring = env->FindClass("java/lang/String");jstring strencode = env->NewStringUTF("GB2312");jmethodID mid = env->GetMethodID(clsstring, "getBytes", "(Ljava/lang/String;)[B");jbyteArray barr = (jbyteArray) env->CallObjectMethod(jstr, mid, strencode);jsize alen = env->GetArrayLength(barr);jbyte *ba = env->GetByteArrayElements(barr, JNI_FALSE);if (alen > 0) {rtn = (char *) malloc(alen + 1);memcpy(rtn, ba, alen);rtn[alen] = 0;}env->ReleaseByteArrayElements(barr, ba, 0);string stemp(rtn);free(rtn);return stemp;
}//后处理找到softmax之后的最大概率值
static int Argmax(float* array, int size) {float max_value = -10000;int max_index = 0;for (int32_t i = 0; i < size; i++) {if (array[i] > max_value) {max_value = array[i];max_index = i;}}return max_index;
}Mat RunInference(string picname){unique_ptr<tflite::FlatBufferModel> model;unique_ptr<tflite::Interpreter> interpreter;model = tflite::FlatBufferModel::BuildFromFile("/sdcard/model/deeplabv3_513.tflite");//我们转换好的模型，我这边直接拷贝到手机里面了，你们可以放到asset文件夹下面model->error_reporter();tflite::ops::builtin::BuiltinOpResolver resolver;tflite::InterpreterBuilder(*model,resolver)(& interpreter);interpreter->AllocateTensors();__android_log_print(ANDROID_LOG_INFO, "mydebug", "Success\n");int input = interpreter->inputs()[0];TfLiteIntArray* dims = interpreter->tensor(input)->dims;int height = dims->data[1];int width = dims->data[2];int channels = dims->data[3];Mat img = imread(picname);//__android_log_print(ANDROID_LOG_INFO, "mydebug", "height %d",img.rows);auto img_inputs = interpreter->typed_tensor<float>(input);//前处理，加赋值给tensorfor(int i = 0;i<img.cols*img.rows*3;i++){img_inputs[i] = (img.data[i]- 128.0)/128.0;}interpreter->Invoke();int output = interpreter->outputs()[0];TfLiteIntArray* output_dims = interpreter->tensor(output)->dims;for(int i = 0;i<4;i++){cout<<output_dims->data[i]<<endl;}float* outputsoftmax = interpreter->typed_output_tensor<float>(0);int* outputlabel = new int[3000*2000];for(int i = 0;i<height*width;i++){outputlabel[i] = Argmax(outputsoftmax+3*i,3);}Mat mat = cv::Mat(3000, 2000, CV_8UC1);for (int i = 0; i < mat.rows; i++){for (int j = 0; j < mat.cols; j++){mat.at<uchar>(i, j) = (uchar)1.0*outputlabel[b] * 100;}}delete[] outputlabel;Mat im_color;applyColorMap(mat, im_color, cv::COLORMAP_JET); //采用colormapresize(im_color,im_color,cv::Size(320,480));return im_color;
}extern "C" JNIEXPORT jintArray JNICALL
Java_com_example_myapplication_MainActivity_decodeFile(JNIEnv* env,jobject /* this */,jstring picname) {string picnam = jstringTostring(env,picname);Mat im_color = RunInference(picnam);int size = im_color.cols * im_color.rows *4;jbyte * outImage = new jbyte[size];__android_log_print(ANDROID_LOG_INFO, "mydebug", "h: %d",im_color.rows);cvtColor(im_color,im_color,CV_RGB2BGRA);jintArray result  = env->NewIntArray(im_color.cols * im_color.rows);env->SetIntArrayRegion(result,0,im_color.cols * im_color.rows,(jint *)im_color.data);return result;
}

安卓java代码

protected void onCreate(Bundle savedInstanceState) {super.onCreate(savedInstanceState);verifyStoragePermissions(this);// Example of a call to a native methodsetContentView(R.layout.activity_main);int[] test = decodeFile("/sdcard/pic/000014_image.png");//我们上面写好的的c++函数，通过JNI调用接口Bitmap result = Bitmap.createBitmap(320,480, Bitmap.Config.RGB_565);result.setPixels(test, 0, 320, 0, 0,320, 480);ImageView imageView = (ImageView) findViewById(R.id.imt_photo);imageView.setImageBitmap(result);//显示返回结果}

好了我们可以看一下效果
这是原图

这个是安卓端的效果图

部署到IOS端

需要MAC环境，在装虚拟机中，未完待续

桌面/PC

linux和windows通用的python

直接贴代码

import tflite_runtime.interpreter as tflite
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt # plt 用于显示图片
import cv2 as cv
interpreter = tf.lite.Interpreter(model_path='/home/william/CLionProjects/DEMO/model/deeplabv3_513.tflite')
interpreter.allocate_tensors()
# Get input and output tensors.
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()
testpic = cv.imread("000014_image.png")
testpic = (1.0*testpic-128)/128
testpic = testpic.reshape(1,3000,2000,3)
testpic = testpic.astype(np.float32)
interpreter.set_tensor(input_details[0]['index'], testpic)
interpreter.invoke()
# The function `get_tensor()` returns a copy of the tensor data.
# Use `tensor()` in order to get a pointer to the tensor.
output_data = interpreter.get_tensor(output_details[0]['index'])
test=output_data.argmax(axis=3)
test = test.reshape(3000,2000)
plt.imshow(test) # 显示图片
plt.axis('off') # 不显示坐标轴
plt.show()

效果

python可以用来实验模型效果，但是实际在工程中使用的时候为了效率还是得用c/c++

部署到linux端

部署到linux比部署到移动端更简单，运行了download_dependencies.sh这个脚本之后，再运行make目录下的build_lib.sh就可以生成再linux上可以运行的.so动态库了
linux端代码和安卓端代码基本上一模一样，安卓端多了一层JNI，linux端不用JNI是纯c++的，代码我就不放出来了，有需要的私聊
这是linux端的效果

部署到windows端

使用Tensorflow lite windows端c++接口是很麻烦的，因为我找了好几天没有找到相关的资料
部署到windows端我并没有使用tensorflow lite，因为刚开始的时候没有尝试移动端和linux端的编译，而是使用的windows作为实验，但是tensorflow官网里面也没有任何tensorflow lite在windows端源码如何编译的资料
虽然windows端的python是可以跑通的，但是如何用c++来调用python又是另一个问题了，而且tensorflow lite的代码本来就是c/c++的，用c++调用python再调用c++这样也有问题，无奈之下，我开始尝试直接使用tensorflow库，这边直接贴代码
在tensorflow/cc的文件夹下面新建一个myinteface的文件夹，然后封装代码和外层接口代码都放在里面
这是tensorflow封装层接口

//这个是tensorflow的推理封装库
#include "direct.h"
#include "tf_inference_lib.h"TfInferenceLib::TfInferenceLib(model_params_t* model_params,tensor_array_t* input_tensors,tensor_array_t* output_tensors): m_input_tensors(input_tensors), m_output_tensors(output_tensors)
{if (NULL == model_params) {MY_ERROR("model params is null \n");}m_model_params = new model_params_t;if (NULL == m_model_params) {MY_ERROR("Alloc local model params error!!!\n");}memcpy(m_model_params, model_params, sizeof(model_params_t));
}TfInferenceLib::~TfInferenceLib()
{delete m_model_params;
}std::string TfInferenceLib::getCurrentModelDir()
{char buffer[256];getcwd(buffer, 256);std::string strDir = buffer;std::cout << "Currrent dir is " << strDir << std::endl;return strDir;
}void TfInferenceLib::TensorDebugInfo(std::string tensor_name, tensorflow::Tensor& tensor)
{std::cout << "tensor name"<< ":" << tensor_name << std::endl;std::cout << "shape:[";int rank = tensor.dims();for (int i = 0; i < rank; i++) {std::cout << tensor.dim_size(i) << ",";}std::cout << "]" << std::endl;tensorflow::DataType eTensorType = tensor.dtype();std::cout << "Tensor type: " << eTensorType << std::endl;std::cout << "Tensor data size: " << tensor.tensor_data().size() << std::endl;std::cout << tensor.SummarizeValue(tensor.NumElements(), true);std::cout << std::endl<< std::endl;
}result_t TfInferenceLib::parseOutputTensors(std::vector<tensorflow::Tensor>& tVecOutputs,tensor_array_t* output_tensor_array)
{assert(tVecOutputs.size() == output_tensor_array->nArraySize);for (int i = 0; i < output_tensor_array->nArraySize; i++) {tensorflow::Tensor cur_tf_tensor = tVecOutputs[i];tensor_t* cur_tensor = &(output_tensor_array->pTensorArray[i]);tensor_params_t* cur_tensor_info = cur_tensor->pTensorInfo;cur_tensor_info->type = (tensor_types_t)cur_tf_tensor.dtype();cur_tensor_info->nElementSize = cur_tf_tensor.NumElements();cur_tensor_info->nDims = cur_tf_tensor.dims();for (int j = 0; j < cur_tensor_info->nDims; j++) {cur_tensor_info->pShape[j] = cur_tf_tensor.dim_size(j);}assert(cur_tensor_info->nLength == cur_tf_tensor.tensor_data().size());memcpy(cur_tensor->pValue, cur_tf_tensor.tensor_data().data(), cur_tensor_info->nLength);}return SUCCESS;
}result_t TfInferenceLib::tfLoadSavedModel()
{tensorflow::Status load_status;std::string strModelAbsolutePath = getCurrentModelDir() + "/" + m_model_params->model_path;if (tensorflow::MaybeSavedModelDirectory(strModelAbsolutePath)) {std::cout << "Saved model is exists, path is " << strModelAbsolutePath << std::endl;} else {std::cout << "Saved model is not exists! path is " << strModelAbsolutePath << std::endl;return FAILED;}//设定显存的参数if (m_model_params->gpu_memory_faction < 0.0001 || m_model_params->gpu_memory_faction > 1.0) {m_model_params->gpu_memory_faction = 0.99;}m_session_options.config.mutable_gpu_options()->set_per_process_gpu_memory_fraction(m_model_params->gpu_memory_faction);m_session_options.config.mutable_gpu_options()->set_allow_growth(true);//Load modelMY_DEBUG("Begin to load mode , path is  %s\n", strModelAbsolutePath.c_str());//load_status = tensorflow::LoadEncSavedModel(//    m_session_options, m_run_options, strModelAbsolutePath,//    { m_model_params->paModelTagSet }, &m_bundle,//    m_model_params->bIsCipher,//    m_model_params->gpu_id);load_status = tensorflow::LoadSavedModel(m_session_options, m_run_options, strModelAbsolutePath,{ m_model_params->paModelTagSet }, &m_bundle);if (load_status.ok()) {std::cout << "Load model succeed!" << std::endl;return SUCCESS;} else {std::cout << "Error load model :" << load_status << std::endl;return MODEL_LOAD_FAILED;}return SUCCESS;
}result_t TfInferenceLib::tfInferenceTensors()
{tensorflow::Status run_status;//拿到模型的函数签名const auto signature_def_map = m_bundle.meta_graph_def.signature_def();const auto signature_def = signature_def_map.at(m_input_tensors->pcSignatureDef);//准备输入Tensor矢量std::vector<tensorflow::Tensor> input_tensor_vec;std::vector<std::string> vecInputTensorNames;for (int j = 0; j < m_input_tensors->nArraySize; ++j) {tensor_t* cur_tensor = &(m_input_tensors->pTensorArray[j]);tensor_params_t* cur_tensor_info = cur_tensor->pTensorInfo;tensorflow::TensorShape cur_tensor_shape;for (int k = 0; k < cur_tensor_info->nDims; ++k) {cur_tensor_shape.AddDim(cur_tensor_info->pShape[k]);}if (cur_tensor_info->type == DT_UINT8) {tensorflow::Tensor input_tensor(tensorflow::DT_UINT8, cur_tensor_shape);memcpy(input_tensor.flat<u8>().data(), cur_tensor->pValue, cur_tensor_info->nLength);input_tensor_vec.push_back(input_tensor);} else if (cur_tensor_info->type == DT_INT32) {tensorflow::Tensor input_tensor(tensorflow::DT_INT32, cur_tensor_shape);memcpy(input_tensor.flat<int>().data(), cur_tensor->pValue, cur_tensor_info->nLength);input_tensor_vec.push_back(input_tensor);} else if (cur_tensor_info->type == DT_FLOAT) {tensorflow::Tensor input_tensor(tensorflow::DT_FLOAT, cur_tensor_shape);memcpy(input_tensor.flat<float>().data(), cur_tensor->pValue, cur_tensor_info->nLength);input_tensor_vec.push_back(input_tensor);} else if (cur_tensor_info->type == DT_STRING) {tensorflow::Tensor input_tensor;std::unique_ptr<tensorflow::TensorProto> upProto(new tensorflow::TensorProto());upProto->set_dtype(tensorflow::DataType::DT_STRING);upProto->add_string_val(cur_tensor->pValue, cur_tensor_info->nLength);upProto->mutable_tensor_shape()->add_dim()->set_size(1);if (!input_tensor.FromProto(*(std::move(upProto)))) {printf("Tensor[%d]::FromProto failed\n", j);return FAILED;}input_tensor_vec.push_back(input_tensor);}vecInputTensorNames.push_back(signature_def.inputs().at(cur_tensor_info->aTensorName).name());}//构造输入pair的矢量std::vector<std::pair<std::string, tensorflow::Tensor>> inputs;for (int i = 0; i < vecInputTensorNames.size(); i++) {inputs.push_back(std::make_pair(vecInputTensorNames[i], input_tensor_vec[i]));}//拿到输出Tensor真实的名字std::vector<std::string> vecWantOutTensorNames;std::vector<tensorflow::Tensor> tVecOutputs;for (int i = 0; i < m_output_tensors->nArraySize; i++) {std::string outputTensorName = m_output_tensors->pTensorArray[i].pTensorInfo->aTensorName;std::string outputTensorInternalName = signature_def.outputs().at(outputTensorName).name();vecWantOutTensorNames.push_back(outputTensorInternalName);}//进行推理MY_DEBUG("Begin to run session!\n");run_status = m_bundle.session->Run(inputs, vecWantOutTensorNames, {}, &tVecOutputs);if (run_status.ok()) {std::cout << "Session run succeed!!!\n";} else {std::cout << "Session run error: " << run_status << std::endl;return FAILED;}//解析输出result_t post_status = parseOutputTensors(tVecOutputs, m_output_tensors);return post_status;
}

这是外层接口

#define DLL_IMPLEMENT
#include <cstring>
#include "my_interface.h"
#include "tf_inference_lib.h"
result_t my_alloc_tensors(tensor_params_array_t* tensors_params,tensor_array_t** tensors)
{MY_CHECK_NULL(tensors_params, PARAM_NULL);tensor_array_t* ptTensorArray = new tensor_array_t;MY_CHECK_NULL(ptTensorArray, MEMORY_MALLOC_FAILED);ptTensorArray->nArraySize = tensors_params->nArraySize;ptTensorArray->pTensorArray = new tensor_t[tensors_params->nArraySize];MY_CHECK_NULL(ptTensorArray->pTensorArray, MEMORY_MALLOC_FAILED);for (int i = 0; i < ptTensorArray->nArraySize; i++) {int nDataSize = 0;tensor_params_t* curTensorParam = &(tensors_params->pTensorParamArray[i]);tensor_t* curTensor = &(ptTensorArray->pTensorArray[i]);switch (curTensorParam->type) {case DT_FLOAT:nDataSize = sizeof(float);break;case DT_DOUBLE:nDataSize = sizeof(double);break;case DT_INT32:nDataSize = sizeof(int);break;case DT_UINT8:nDataSize = sizeof(u8);break;case DT_STRING:nDataSize = sizeof(s8);break;case DT_BOOL:nDataSize = sizeof(bool);break;default:nDataSize = 1;break;}curTensorParam->nElementSize = 1;for (int j = 0; j < curTensorParam->nDims; j++) {curTensorParam->nElementSize *= curTensorParam->pShape[j];}curTensorParam->nLength = curTensorParam->nElementSize * nDataSize;curTensor->pValue = new u8[curTensorParam->nLength];MY_CHECK_NULL(curTensor->pValue, MEMORY_MALLOC_FAILED);curTensor->pTensorInfo = new tensor_params_t;memcpy(curTensor->pTensorInfo, curTensorParam, sizeof(tensor_params_t));}strcpy(ptTensorArray->pcSignatureDef, tensors_params->pcSignatureDef);*tensors = ptTensorArray;return SUCCESS;
}
result_t release_tensors(tensor_array_t* tensors)
{MY_CHECK_NULL(tensors, PARAM_NULL);for (int i = 0; i < tensors->nArraySize; i++) {tensor_t* curTensor = &(tensors->pTensorArray[i]);delete[]((u8*)(curTensor->pValue));delete (curTensor->pTensorInfo);}delete[] tensors->pTensorArray;delete tensors;return SUCCESS;
}
/*** 功能： 申请输入/输出tensor array的内存* 参数：*     input_tensors_params（in） ： 输入的tensor参数结构体；*     output_tensors_params(in) : 输出的tensor参数结构体；*     input_tensors（out) :        申请的输入tensor数组；*     output_tensors（out) :       申请的输出tensor数组；**/
DLL_API result_t init_tensors(tensor_params_array_t* input_tensors_params,tensor_params_array_t* output_tensors_params,tensor_array_t** input_tensors,tensor_array_t** output_tensors)
{result_t result = SUCCESS;//分配输入Tensor数组内存result = my_alloc_tensors(input_tensors_params, input_tensors);if (result != SUCCESS) {MY_ERROR("Alloc input tensors error !!!\n");}//分配输出Tensor数组内存result = my_alloc_tensors(output_tensors_params, output_tensors);if (result != SUCCESS) {MY_ERROR("Alloc input tensors error !!!\n");}return result;
}
/*** 功能： 释放申请的tensor array的内存* 参数*     input_tensors（in) :        申请的输入tensor数组指针；*     output_tensors（in) :       申请的输出tensor数组指针；**/
DLL_API result_t deinit_tensors(tensor_array_t* input_tensors,tensor_array_t* output_tensors)
{result_t res = SUCCESS;res = release_tensors(input_tensors);if (res != SUCCESS) {MY_ERROR("Release input tensors error!!!\n");}res = release_tensors(output_tensors);if (res != SUCCESS) {MY_ERROR("Release output tensors error!!!\n");}return SUCCESS;
}
result_t gpu_card_visible(char* visible_card_id_list)
{char env_name[] = "CUDA_VISIBLE_DEVICES";const char* new_cuda_value = visible_card_id_list;char* old_cuda_value = getenv(env_name);if (NULL == old_cuda_value) {old_cuda_value = "";}std::cout << "old cuda value is :" << old_cuda_value << std::endl;
#ifdef _WIN32char tmp_env_exp[256] = { 0 };sprintf(tmp_env_exp, "%s=%s", env_name, new_cuda_value);putenv(tmp_env_exp);
#elsesetenv(env_name, new_cuda_value, 1);
#endifchar* new_seted_cuda_value = getenv(env_name);std::cout << "New setted cuda value is :" << new_seted_cuda_value;return SUCCESS;
}
/*** 功能： 根据模型参数装载tensorflow模型* 参数：*     model_param（in) : 模型的输入参数*     input_tensors（in) :        输入tensor数组指针；*     output_tensors（in) :       输出tensor数组指针；*     model_handle（out) :装载好的模型句柄 **/
DLL_API result_t load_model(model_params_t* load_model_param,tensor_array_t* input_tensors,tensor_array_t* output_tensors,model_handle_t* load_model_handle)
{TfInferenceLib* tfInferenceLib = new TfInferenceLib(load_model_param, input_tensors, output_tensors);gpu_card_visible(load_model_param->visibleCard);tfInferenceLib->tfLoadSavedModel();load_model_handle->model_handle = tfInferenceLib;return SUCCESS;
}
/*** 功能： 释放模型的内存* 参数：*      model_handle（in):要释放的模型句柄
**/
DLL_API result_t release_model(model_handle_t* load_model_handle)
{TfInferenceLib* tfInferenceLib = (TfInferenceLib*)load_model_handle->model_handle;delete tfInferenceLib;return SUCCESS;
}
/***  功能：进行推理，推理后的结果放到output_tensors中*  参数：*       model_handle(in) 模型句柄**/
DLL_API result_t inference_tensors(model_handle_t* load_model_handle)
{TfInferenceLib* tfInferenceLib = (TfInferenceLib*)load_model_handle->model_handle;result_t res = tfInferenceLib->tfInferenceTensors();return res;
}

然后修改tensorflow/cc文件夹下面的BUILD文件
在最后加上

c_library(name = "my_tensorflow",srcs = ["my_inference/common.h","my_inference/my_interface.cc","my_inference/my_interface.h","my_inference/tf_inference_lib.cc","my_inference/tf_inference_lib.h",],#linkshared = 1,deps = [":cc_ops",":client_session",":coordinator",":queue_runner",":scope","//tensorflow/cc/saved_model:constants","//tensorflow/cc/saved_model:loader","//tensorflow/cc/saved_model:signature_constants","//tensorflow/cc/saved_model:tag_constants","//tensorflow/core:core_cpu","//tensorflow/core:framework","//tensorflow/core:lib","//tensorflow/core:lib_internal","//tensorflow/core:protos_all_cc","//tensorflow/core:tensorflow",],
)

然后再修改tensorflow的BUILD文件

tf_cc_shared_object(name = "tensorflow_cc",linkopts = select({"//tensorflow:macos": ["-Wl,-exported_symbols_list,$(location //tensorflow:tf_exported_symbols.lds)",],"//tensorflow:windows": [],"//conditions:default": ["-z defs","-Wl,--version-script,$(location //tensorflow:tf_version_script.lds)",],}),per_os_targets = True,soversion = VERSION,visibility = ["//visibility:public"],# add win_def_file for tensorflow_ccwin_def_file = select({# We need this DEF file to properly export symbols on Windows"//tensorflow:windows": ":tensorflow_filtered_def_file","//conditions:default": None,}),deps = ["//tensorflow:tf_exported_symbols.lds","//tensorflow:tf_version_script.lds","//tensorflow/c:c_api","//tensorflow/c/eager:c_api","//tensorflow/cc:cc_ops","//tensorflow/cc:client_session","//tensorflow/cc:scope","//tensorflow/cc/profiler","//tensorflow/cc:my_tensorflow",             #这是我们添加的接口"//tensorflow/core:tensorflow",] + if_ngraph(["@ngraph_tf//:ngraph_tf"]),
)

用bazel编译tensorflow的tensorflow_cc，顺利的话我们就可以编译出来tensorflow_cc.dll和tensorflow_cc.lib

然后我们在我们自己的工程里就可以引入动态库和静态库调用我们的接口了
最外一层还得加一个总的decodeFile的接口我还没时间写

调用接口实现，下面是我的一个例子

#include <stdio.h>
#include <string.h>
#include <assert.h>
#include "opencv2/opencv.hpp"
#include "DCS_DeepLearningRegionDetection.h"#define BATH_SIZE 1int main(int argc, char **argv)
{//申请输入输出内存tensor_params_array_t in_tensor_params_ar = { 0 };tensor_params_array_t out_tensor_params_ar = { 0 };tensor_array_t *input_tensor_array = NULL;tensor_array_t *output_tensor_array = NULL;//输入Tensor数组参数设置in_tensor_params_ar.nArraySize = 1;strcpy(in_tensor_params_ar.pcSignatureDef, "predict_images");in_tensor_params_ar.pTensorParamArray = (tensor_params_t *)malloc(in_tensor_params_ar.nArraySize * sizeof(tensor_params_t));tensor_params_t *cur_in_tensor_params = &(in_tensor_params_ar.pTensorParamArray[0]);cur_in_tensor_params->nDims = 4;cur_in_tensor_params->type = DT_UINT8;cur_in_tensor_params->pShape[0] = BATH_SIZE;cur_in_tensor_params->pShape[1] = 3000;  //Hcur_in_tensor_params->pShape[2] = 2000; //Wcur_in_tensor_params->pShape[3] = 3;    //channelstrcpy(cur_in_tensor_params->aTensorName, "ImageTensor");//输出Tensor数组参数设置out_tensor_params_ar.nArraySize = 1;strcpy(out_tensor_params_ar.pcSignatureDef, "predict_images");out_tensor_params_ar.pTensorParamArray = (tensor_params_t *)malloc(out_tensor_params_ar.nArraySize * sizeof(tensor_params_t));tensor_params_t *cur_out_tensor_params0 = &(out_tensor_params_ar.pTensorParamArray[0]);cur_out_tensor_params0->type = DT_INT32;cur_out_tensor_params0->nDims = 3;cur_out_tensor_params0->pShape[0] = BATH_SIZE;cur_out_tensor_params0->pShape[1] = 3000;cur_out_tensor_params0->pShape[2] = 2000;strcpy(cur_out_tensor_params0->aTensorName, "SemanticPredictions");//调用API申请Tensor数组内存if (SUCCESS != init_tensors(&in_tensor_params_ar, &out_tensor_params_ar,&input_tensor_array, &output_tensor_array)){printf("Open tensor memory error\n");}//设置模型加载参数model_params_t tModelParams = { 0 };model_handle_t tModelHandel = { 0 };tModelParams.cpu_or_gpu = 0;strcpy(tModelParams.visibleCard, "0");//strcpy(tModelParam.visibleCard, "0,1");tModelParams.gpu_id = 0;tModelParams.gpu_memory_faction = 0.9;//tModelParams.bIsCipher = true;//strcpy(tModelParams.model_path, "models/object_detection_enc/1");tModelParams.bIsCipher = false;strcpy(tModelParams.model_path, "models/test/1");strcpy(tModelParams.paModelTagSet, "serve");//调用API装载模型if (load_model(&tModelParams, input_tensor_array, output_tensor_array, &tModelHandel) != SUCCESS){printf("Load Model error!!!\n");}cv::Mat bgrImage, rgbImage;bgrImage = cv::imread("test_data/000014_image.png");cv::cvtColor(bgrImage, rgbImage, cv::COLOR_BGR2RGB);int img_size = rgbImage.rows * rgbImage.cols * rgbImage.channels();tensor_t *cur_input_tensor = &(input_tensor_array->pTensorArray[0]);tensor_params_t *cur_input_tensor_info = cur_input_tensor->pTensorInfo;std::cout << "Cur tensor value length: " << cur_input_tensor_info->nLength << std::endl;assert(img_size == cur_input_tensor_info->nLength);memcpy(cur_input_tensor->pValue, rgbImage.ptr<unsigned char>(0), img_size);printf("Call api to inferencing.....\n");inference_tensors(&tModelHandel);printf("End inference!!!\n");//打印推理结果tensor_t * cur_output_tensor_class = &(output_tensor_array->pTensorArray[0]);int *seg = (int*)cur_output_tensor_class->pValue;cv::Mat mat = cv::Mat(3000, 2000, CV_8UC1);int b = 0;for (int i = 0; i < mat.rows; i++){for (int j = 0; j < mat.cols; j++){mat.at<uchar>(i, j) = (uchar)1.0*seg[b] * 100;b++;}}cv::Mat im_color;cv::applyColorMap(mat, im_color, cv::COLORMAP_JET);//释放申请的Tensor数组内存deinit_tensors(input_tensor_array, output_tensor_array);release_model(&tModelHandel);free(in_tensor_params_ar.pTensorParamArray);free(out_tensor_params_ar.pTensorParamArray);system("pause");
}

这是window端Debug下用imagewatch看的效果

window端 tflite

tensorflow官方没有提供tflite的windows版本，在GitHub上找到了别人实现的window版本
https://github.com/qintao97/tensorflow_lite

未完待续，最后还剩IOS了