快速构建纺织品缺陷分类检测方案

一、用例介绍

该用例使用OpenVINO工具包包含的推理引擎,显示了如何在工业质量检验应用中利用分类网络的示例。该用例提供了一种支持人工智能的方法来将来自织物检查摄像机的输入帧分类为缺陷或良好。

二、安装环境

  • 硬件要求: 英特尔处理器
  • 系统要求: Ubuntu18.04 LTS
  • OpenVINO: 2021.2
  • python: 3.6
  • 安装用时: 30min

三、工作原理

  • 1.纺织品检验视频:

用于模拟纺织品检查摄像机的示例视频。该视频包括五种常见的纺织品缺陷,其中一些不容易被人眼注意到。

  • 2.推理模块:

从视频中提取帧,并执行推论,将帧归类为有缺陷或没有缺陷。每个帧的分类速率也记录并在终端输出显示

  • 3.它的工作流如下

首先我们在纺织品图片标签数据,并拆分为训练数据集和测试数据集,接着用训练数据集来训练模型,在模型优化器中进行优化生成IR文件,然后在推理引擎上加载模型和测试数据集,最后我们可以在工业生产中完成部署。

20210408163631

四、安装步骤

  • 1.解压下载的包并进入解压目录,更改edgesoftware文件的权限 chmod +x edgesoftware

  • 2.运行命令,安装 ./edgesoftware install

2021-04-08-09-42-06

  • 3.会提示输入秘钥,我们输入下载时保存好的key就可以了。过程中会提示配置pip和apt的国内加速源,点击yes,按图配置即可。

2021-04-08-09-43-04

  • 4.配置完成后我们可看到相关配置,如下图,apt同理

2021-04-08-09-43-56

  • 5.安装完成

2021-04-08-10-28-39

五、运行使用

  • 1.进入工作路径

2021-04-08-10-36-28

  • 2.查看初始化脚本

2021-04-08-10-36-39

  • 3.激活Python虚拟环境

这里遇到一个报错,运行脚本后并没有进入虚拟环境

2021-04-08-10-39-02

这是因为脚本没有执行权限,下图是解决办法,可以看到已经成功进入m_classf环境中

2021-04-08-10-39-15

  • 4.我们需要下载python的一些依赖包

2021-04-08-10-50-14

  • 5.该纺织品包含的样本数据有五种类型的缺陷,如下图

20210408171718

  • 6.接下来我们在单个测试图像上运行推理

python textile/inference.py --model models/model.xml --label models/labels.txt --image data/test_data/color_flecks/frame_a.png --device cpu

output1

  • 7.在一组图像上运行推理

python textile/inference.py --model models/model.xml --label models/labels.txt --dir data/test_data/hole --device cpu

output2

  • 8.计算图像测试数据集上的混淆矩阵

python textile/inference.py --model models/model.xml --label models/labels.txt --test_dataset data/test_data/ --confusion_matrix --device cpu

这里执行代码的时候遇到一个报错,说没有写入权限

2021-04-09-17-39-09

解决办法是chmod添加权限

2021-04-09-17-48-26

再次运行就ok了

output3

运行完成后输入deactivate可以退出虚拟环境

附用例网址:https://software.intel.com/content/www/us/en/develop/articles/textile-defect-classifier.html

相关文章

功能概要:(目前已实现功能)公共展示部分:1.网站首页展示...
大体上把Python中的数据类型分为如下几类: Number(数字) ...
开发之前第一步,就是构造整个的项目结构。这就好比作一幅画...
源码编译方式安装Apache首先下载Apache源码压缩包,地址为ht...
前面说完了此项目的创建及数据模型设计的过程。如果未看过,...
python中常用的写爬虫的库有urllib2、requests,对于大多数比...