[发明专利]一种基于异构计算的图像预处理方法及系统

说明书

本发明提供了计算机技术领域的一种基于异构计算的图像预处理方法及系统，方法包括如下步骤：步骤S1、将采集的彩色图像保存至CPU内存，再将CPU内存保存的所述彩色图像保存至GPU显存；步骤S2、将GPU显存保存的所述彩色图像转换成RGB24图像；步骤S3、在GPU显存内对所述RGB24图像进行归一化计算，得到归一化数据；步骤S4、利用AI目标检测算法检测所述归一化数据中的目标数据，将所述目标数据传输至CPU内存并进行绘制。本发明的优点在于：极大的提升了图像预处理的效率以及稳定性。

技术领域

本发明涉及计算机技术领域，特别指一种基于异构计算的图像预处理方法及系统。

背景技术

图像在采集完成后，需要进行一系列的预处理再进行输出，以保证图像输出的质量。针对图像的预处理，传统上是在单一CPU架构上进行计算的，即将图像采集到CPU的内存中，CPU读取内存并进行图像预处理。但是，传统的方法对于工业高速(譬如：10ms内)的检测应用场景来说，CPU处理效率无法满足要求，而且CPU负载太高会导致系统不稳定。

因此，如何提供一种基于异构计算的图像预处理方法及系统，实现提升图像预处理的效率以及稳定性，成为一个亟待解决的技术问题。

发明内容

本发明要解决的技术问题，在于提供一种基于异构计算的图像预处理方法及系统，实现提升图像预处理的效率以及稳定性。

第一方面，本发明提供了一种基于异构计算的图像预处理方法，包括如下步骤：

步骤S1、将采集的彩色图像保存至CPU内存，再将CPU内存保存的所述彩色图像保存至GPU显存；

步骤S2、将GPU显存保存的所述彩色图像转换成RGB24图像；

步骤S3、在GPU显存内对所述RGB24图像进行归一化计算，得到归一化数据；

步骤S4、利用AI目标检测算法检测所述归一化数据中的目标数据，将所述目标数据传输至CPU内存并进行绘制。

进一步地，所述步骤S1具体为：

通过相机采集彩色图像，将采集的所述彩色图像保存至CPU内存，再将CPU内存保存的所述彩色图像通过DMA传输并保存至GPU显存。

进一步地，所述步骤S2具体为：

通过插值法将GPU显存保存的所述彩色图像转换成RGB24图像。

进一步地，所述步骤S3具体为：

GPU对GPU显存内求取所述RGB24图像的均值和方差，将所述RGB24图像的每个像素值减去均值后再除以方差，得到归一化数据。

进一步地，所述步骤S4具体为：

GPU利用AI目标检测算法检测所述归一化数据中包括目标类别、相似度以及目标顶点坐标的目标数据，将所述目标数据通过DMA传输至CPU内存并进行绘制和显示。

第二方面，本发明提供了一种基于异构计算的图像预处理系统，包括如下模块：

彩色图像保存模块，用于将采集的彩色图像保存至CPU内存，再将CPU内存保存的所述彩色图像保存至GPU显存；

彩色图像转换模块，用于将GPU显存保存的所述彩色图像转换成RGB24图像；

归一化计算模块，用于在GPU显存内对所述RGB24图像进行归一化计算，得到归一化数据；

目标检测模块，用于利用AI目标检测算法检测所述归一化数据中的目标数据，将所述目标数据传输至CPU内存并进行绘制。

进一步地，所述彩色图像保存模块具体为：