一种基于5G边缘计算技术的图像识别方法技术

本发明专利技术涉及图像识别技术领域，具体是一种基于5G边缘计算技术的图像识别方法，包括如下步骤：通过现场端的图像采集单元获取图像，并对图像进行预处理；通过图像识别单元对预处理后的图像进行识别，并将图像识别结果传输给5G通信模块，其中，所述图像识别单元设置于现场端；5G通信模块将识别结果传输给云端服务器。本发明专利技术将图像识别单元部署于靠近数据源头的现场端，现场端直接对图像进行识别，降低了对通信的要求并缩短了识别的时耗，使识别更加高效，适宜进一步推广应用。适宜进一步推广应用。

【技术实现步骤摘要】
一种基于5G边缘计算技术的图像识别方法


[0001]本专利技术涉及图像识别
，具体是一种基于5G边缘计算技术的图像识别方法。

技术介绍

[0002]边缘计算，是指在靠近物或数据源头的一侧，采用网络、计算、存储、应用核心能力为一体的开放平台，就近提供最近端服务。其应用程序在边缘侧发起，产生更快的网络服务响应，满足行业在实时业务、应用智能、安全与隐私保护等方面的基本需求。
[0003]现有的神经网络图像识别模型往往部署在云端，现场采集图像后需先上传至云端，再进行识别，对通信要求高且较为繁琐耗时。此外，现有图像识别方法大多是对待识别图像进行缩放，然后再将缩放后的图像输入到图像模型中进行识别。当待识别图像中目标较小，背景信息过多时，识别的准确率较低且计算量也会较大。

技术实现思路

[0004]针对上述问题，本专利技术提出一种基于5G边缘计算技术的图像识别方法。
[0005]本专利技术通过以下技术方案予以实现：
[0006]一种基于5G边缘计算技术的图像识别方法，包括如下步骤：
[0007]通过现场端的图像采集单元获取图像，并对图像进行预处理；
[0008]通过图像识别单元对预处理后的图像进行识别，并将图像识别结果传输给5G通信模块，其中，所述图像识别单元设置于现场端；
[0009]5G通信模块将识别结果传输给云端服务器。
[0010]进一步的，所述图像识别单元中加载有用于识别图像的神经网络模型，神经网络模型对预处理后的图像进行识别，并输出识别结果。
[0011]进一步的，所述用于识别图像的神经网络模型为知识蒸馏改进的SimpleCNN算法，具体为：以训练完成的VGG网络指导未完全训练的SimpleCNN算法进行训练，得到轻量化的神经网络模型。
[0012]进一步的，所述用于识别图像的神经网络模型的训练方法，包括：
[0013]将图像数据集作为输入，通过VGG19网络对SimpleCNN网络进行蒸馏，通过设置不同的蒸馏温度，得到不同的损失函数后进行反向传播，然后将模型参数保存，得到用于识别图像的神经网络模型。
[0014]进一步的，将图像数据集作为输入，通过VGG19网络对SimpleCNN网络进行蒸馏，通过设置不同的蒸馏温度，得到不同的损失函数后进行反向传播，然后将模型参数保存，得到用于识别图像的神经网络模型，具体包括如下步骤：
[0015]计算数据集中各图像的像素均值，并将每张图像调整为224*224像素；
[0016]将调整后图像的各通道像素值减去上述求得的像素均值，得到图像处理数据；
[0017]构建知识蒸馏模块，知识蒸馏模块由已训练好参数的VGG19网络以及未经训练的
SimpleCNN网络组成；
[0018]通过VGG19网络指导SimpleCNN网络进行训练；训练SimpleCNN时，图像分批输入，对每批次图像进行3D卷积、批标准化、最大池化、ReLU、2DLSTM操作得到特征图像；
[0019]将特征图像进行多通道展平，并通过Dropout层去除不必要神经元，线性映射加深网络深度，最后通过Softmax分类输出，得到损失函数和特征数据；
[0020]将整个网络的损失函数替换为知识蒸馏各软目标与硬目标的线性加权，将损失函数进行反向传播，不断进行参数修正，得到深层特征；
[0021]将验证集图像输入神经网络中，将得到的结果与图像标签进行比对运算，计算误差，根据误差进行反向计算，更新模型参数，最后将模型参数保存，得到用于识别图像的神经网络模型。
[0022]进一步的，将整个网络的损失函数替换为知识蒸馏各软目标与硬目标的线性加权，将损失函数进行反向传播，具体包括如下步骤：
[0023]网络训练时，VGG19设置蒸馏温度T1，生成软目标；SimpleCNN网络分别设置蒸馏温度T1’
和T2，分别生成软目标A和软目标B，其中T1＝T1’
≠T2；硬目标为标签值；将软目标和软目标A取KL散度得到蒸馏损失，将软目标B和硬目标取交叉熵损失得到误差损失，将蒸馏损失、误差损失按比例进行线性加权得到最终损失，进行反向传播。
[0024]进一步的，图像预处理，具体包括如下步骤：
[0025]1)对原图像进行灰度处理；
[0026]2)对灰度图像进行序贯滤波处理，以除去灰度图中的噪声；
[0027]3)通过边缘算子识别灰度图中边界信息，并确定图像中的ROI区域；
[0028]4)根据灰度图中的边界信息，沿边界从原图像中截取出对应ROI区域；
[0029]5)将截取出的ROI区域缩放至固定尺寸，接着将其粘贴于预设尺寸的白色空白底板图片中。
[0030]进一步的，步骤1)具体步骤如下：
[0031]将各图像R、G、B三通道的通道值进加权平均运算并将运算结果赋值作为该点的灰度值。
[0032]优选的，步骤2)中边缘算子选用sobel、Roberts、Prewitt、Canny中任意一种。
[0033]在具有相同硬件基础的情况下，本专利技术方法具有一定的通配性，因此，本专利技术还提供一种计算机可读的存储介质，所述的存储介质中存储有至少一条指令、至少一段程序、代码集或指令集，所述的至少一条指令、至少一段程序、代码集或指令集由处理器加载并执行实现上述基于5G边缘计算技术的图像识别方法。
[0034]与现有技术相比，本专利技术具有以下有益效果：
[0035]1)本专利技术提供的基于5G边缘计算技术的图像识别方法将图像识别单元部署于靠近数据源头的现场端，现场端直接进行图像的识别，降低了对通信的要求并缩短了识别的时耗，使识别更加高效。
[0036]2)本专利技术提供的知识蒸馏改进的SimpleCNN网络能够减少数据数量的需求，加强了网络的泛华能力，同时使网络模型更加轻量化，适于现场端的部署使用。
[0037]3)本专利技术提供的图像预处理方法能够对图片中ROI区域进行高效提取，避免了背景信息过多对图像识别算法的影响，有利于提升图像识别的准确率与效率。
附图说明
[0038]为了更清楚地说明本专利技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本专利技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。
[0039]图1为本专利技术的简要流程示意图。
具体实施方式
[0040]为使本专利技术实施方式的目的、技术方案和优点更加清楚，下面对本专利技术实施方式中的技术方案进行清楚、完整地描述，显然，所描述的实施方式是本专利技术一部分实施方式，而不是全部的实施方式。基于本专利技术中的实施方式，本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施方式，都属于本专利技术保护的范围。基于本专利技术中的实施方式，本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施方式，本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于5G边缘计算技术的图像识别方法，其特征在于，包括如下步骤：通过现场端的图像采集单元获取图像，并对图像进行预处理；通过图像识别单元对预处理后的图像进行识别，并将图像识别结果传输给5G通信模块，其中，所述图像识别单元设置于现场端；5G通信模块将识别结果传输给云端服务器。2.根据权利要求1所述的一种基于5G边缘计算技术的图像识别方法，其特征在于，所述图像识别单元中加载有用于识别图像的神经网络模型，神经网络模型对预处理后的图像进行识别，并输出识别结果。3.根据权利要求2所述的一种基于5G边缘计算技术的图像识别方法，其特征在于，用于识别图像的神经网络模型为知识蒸馏改进的SimpleCNN算法，具体为：以训练完成的VGG网络指导未完全训练的SimpleCNN算法进行训练，得到轻量化的神经网络模型。4.根据权利要求3所述的一种基于5G边缘计算技术的图像识别方法，其特征在于，所述用于识别图像的神经网络模型的训练方法，包括：将图像数据集作为输入，通过VGG19网络对SimpleCNN网络进行蒸馏，通过设置不同的蒸馏温度，得到不同的损失函数后进行反向传播，然后将模型参数保存，得到用于识别图像的神经网络模型。5.根据权利要求4所述的一种基于5G边缘计算技术的图像识别方法，其特征在于，将图像数据集作为输入，通过VGG19网络对SimpleCNN网络进行蒸馏，通过设置不同的蒸馏温度，得到不同的损失函数后进行反向传播，然后将模型参数保存，得到用于识别图像的神经网络模型，具体包括如下步骤：计算数据集中各图像的像素均值，并将每张图像调整为224*224像素；将调整后图像的各通道像素值减去上述求得的像素均值，得到图像处理数据；构建知识蒸馏模块，知识蒸馏模块由已训练好参数的VGG19网络以及未经训练的SimpleCNN网络组成，通过VGG19网络指导SimpleCNN网络进行训练；训练SimpleCNN时，图像分批输入，对每批次图像进行3D卷积、批标准化、最大池化、ReLU、2DLSTM操作得到特征图像；将特征图像进行多通道展平，并通过Dropout层去除不必要神经元，线性映射加深网络深度，最后通过Softmax分类输出，得到损失...

【专利技术属性】
技术研发人员：缪品章，陈苹，缪文雄，翁鲲鹏，
申请(专利权)人：富春科技股份有限公司，
类型：发明
国别省市：