流水线形式做机器视觉开发

2020-07-264205
AI 快速预览详细 收起
本文介绍了使用TensorFlow Object Detection API和OpenVINO结合NCS2进行机器视觉识别开发的技术路线。通过预训练模型迁移学习生成特定场景的训练模型,并利用软硬件结合减轻边缘计算端的主控硬件要求。适用于LattePanda Delta等配置。文章详细介绍了从安装环境到模型训练、优化和应用代码开发的七个步骤,帮助开发者快速实现机器视觉识别。


目前快速开发可实际部署的机器视觉识别技术路线是什么?

答案是:TensorFlow Object Decetion API+OpenVINO&NCS2
原因是什么?
1.TensorFlow Object Decetion API可快速利用成熟预训练模型结合少量的数据集就能进行迁移学习生成特定场景的训练模型
2.OpenVINO&NCS2通过软硬件结合大大减轻了边缘计算端的主控硬件要求,使得类似LattePanda Delta的配置也能轻松运行。


有了他们,我们就可以流水线形式开发机器视觉识别的使用场景方案。
步骤:
1.安装TensorFlow Object Decetion API+OpenVINO环境
2.数据集采集:

对使用场景的物体拍照建立待训练数据集
流水线形式做机器视觉开发图1

流水线形式做机器视觉开发图2流水线形式做机器视觉开发图3

流水线形式做机器视觉开发图4流水线形式做机器视觉开发图5

3.数据集标注:
使用图片标注软件标注图片
流水线形式做机器视觉开发图6流水线形式做机器视觉开发图7


4.模型训练:
下载预训练模型,使用上个步骤的图片标注数据集进行迁移学习训练,训练完成得到TensorFlow冻结图模型pb文件。模型训练时间较长,看机器配置及训练参数,啥时NCS3能支持训练就好了:)
流水线形式做机器视觉开发图8


5.使用测试图片测试训练的模型:
流水线形式做机器视觉开发图9流水线形式做机器视觉开发图10


6.openvino模型优化:
通过模型优化器Model Optimizer把pb文件优化为openvino能读取的IR文件
流水线形式做机器视觉开发图11


7.IE推理引擎及应用代码开发:
用C++或python开发IE推理引擎及应用代码,如果对执行效率要求比较高建议C++开发。
基本开发过程:1.载入硬件插件 2.读取IR文件 3.配置输入输出 4.载入模型 5.创建infer request 【6.准备输入数据 7.执行推理计算 8.处理推理输出 9.根据输出应用代码处理】
流水线形式做机器视觉开发图12



流水线形式做机器视觉开发图13

......

经过以上7个步骤已经可以流水线形式进行机器视觉识别方案开发。如果应用场景、逻辑一样,只是视觉处理的物体不一样,那么代码都不用改,只需把2-6步骤按照新物体做一遍就可以跑了。流水线开发机器视觉识别不是梦想了:)

202007266044..png (430 Bytes, 下载次数: 2838)

机器视觉开发 · TensorFlow+OpenVINO流水线方案_image_3.webp

创作许可协议

本项目采用 None(不开放任何权利,保留所有权利) 进行许可。

评论(0)
- 没有更多了 -