基于Tensorflow Lite的CIFAR10图像分类模型部署

本博客开设了”深度学习计算机视觉实战“专栏,有兴趣的朋友欢迎访问并分享给您的朋友。
欢迎关注公众号”计算机视觉与OpenCV“,本号可以获取一些学习资料,而且不定期会发出赠书活动。
欢迎加入QQ群”187042448“获取更多的软件编程、AI、机器学习、深度学习的资料。
欢迎加微信study428拉入微信群,暂时VI和VII群还可以扫码进群(CSDN图片没法调整大小,我就把三个图都拼到一起大家看看了。)

本文分享的内容来源于《深度学习计算机视觉实战》一书,该书由刘东研究员和肖铃合作完成,由电子工业出版社出版。
该书从算法导读到模型训练,到模型部署一站式搞定,书中案例注释详细,均已通过运行验证。该书包括4个部分:
第一部分(1~2章)深度学习和计算机视觉的基础和算法讲解;
第二部分(3~6章)传统图像处理算法的案例讲解;
第三部分(7-11章)计算机视觉方向的案例讲解;
第四部分(12~13章)Tensorflow Lite讲解(源码分析、模型优化、模型转换等)以及使用Tensorflow Lite在PC端和移动端的部署案例讲解。

本书可以为计算机视觉入门的读者和想要对模型进行工程部署的读者提供参考与帮助。京东链接:https://u.jd.com/rwk3HPT
在这里插入图片描述

本节分享的内容为Windows端的模型部署,如果读者朋友想要使用移动端和Linux端的模型部署,建议参考书中案例。

如下PPT的配套视频可以去这里观看:
https://www.bilibili.com/video/BV1yb4y1y7TR/
下面结合视频中使用的PPT,讲述一下Windows端的模型部署。
在这里插入图片描述

内容主要包括以下四个方面:
在这里插入图片描述

这是本书的图书展示,前面有介绍不做过多说明。
在这里插入图片描述

模型部署之前,需要将模型转换为Tensorflow Lite推理框架可以读取的文件格式,即.tflite格式。
模型转换请参考文章:Tensorflow Lite模型部署之模型转换
在这里插入图片描述

模型量化属于模型优化的一部分,其中量化感知训练是在模型训练过程中的量化,训练后量化是在模型训练完成后进行的量化。
在这里插入图片描述

在模型部署的过程中,需要依赖Tensorflow Lite,在PC端是将其编译为二进制库文件进行链接,因为Tensorflow Lite在2.4版本之后才提供CMakeLists.txt编译脚本,所以使用CMake进行编译的过程中还有一些问题,本次的讲解使用的是版本2.6。
在这里插入图片描述

下载了源码,编译生成项目工程文件。
在这里插入图片描述

使用Visual Studio 2019打开工程进行编译。
在这里插入图片描述

因为项目配置的问题,编译中还是有一些问题需要手动去处理。
在这里插入图片描述
在这里插入图片描述

处理完错误之后终于可以生成二进制库文件了,不过这里生成的是Debug版本的动态库,如果需要生成Release版本的,需要手动修改,并重新编译项目。
在这里插入图片描述

右边图中的文件夹是部署CIFAR10图像分类的代码,模型的训练方法在书中有案例讲解,模型转换方法也有讲解,测试代码在书中有详细展示并注释,这里就是编译的脚本文件。
在这里插入图片描述

如下是设置使用的头文件。
在这里插入图片描述

使用CMake软件编译测试项目。
在这里插入图片描述

代码错误处理如下:
在这里插入图片描述

头文件不全导致代码中出现了变量未定义的问题。
在这里插入图片描述

补充后编译运行,即可进行正确的推理。
在这里插入图片描述

测试案例的代码可以在书中参考。

猜你喜欢

转载自blog.csdn.net/lxiao428/article/details/120785813