[发明专利]图像处理方法、装置、电子设备及计算机可读存储介质

说明书

本申请公开了一种图像处理方法、装置、电子设备及计算机可读存储介质，涉及图像处理技术领域。具体实现方案为：获取待处理图像；将所述待处理图像转化为三通道YUV图像；对所述三通道YUV图像中的Y通道图像、U通道图像和V通道图像进行卷积运算，分别生成R通道图像、G通道图像和B通道图像，得到三通道RGB图像；对所述三通道RGB图像进行预处理。根据本申请中方案，可以提升图像预处理速度。

技术领域

本申请涉及图像处理技术领域，尤其涉及一种图像处理方法、装置、电子设备及计算机可读存储介质。

背景技术

在现有的计算机视觉技术中，图像预处理是必不可少的一步。目前常用的图像预处理方式为：针对待处理图像中的每一个像素点，依次进行数据格式整理、RGB转化、数据归一化等流程。但此方式会耗费较多的处理时间，造成处理速度慢。

发明内容

本申请实施例提供一种图像处理方法、装置、电子设备及计算机可读存储介质，以解决现有图像预处理方式的处理速度慢的问题。

为了解决上述技术问题，本申请是这样实现的：

第一方面，本申请实施例提供了一种图像处理方法，包括：

获取待处理图像；

将所述待处理图像转化为三通道YUV图像；

对所述三通道YUV图像中的Y通道图像、U通道图像和V通道图像进行卷积运算，分别生成R通道图像、G通道图像和B通道图像，得到三通道RGB图像；

对所述三通道RGB图像进行预处理。

这样，在图像预处理过程中，可以借助卷积运算对转化得到的三通道YUV图像中的Y通道图像、U通道图像和V通道图像进行处理，以生成三通道RGB图像，并对该三通道RGB图像进行预处理，从而缩短将YUV图像转化为RGB图像的时间，从而提升图像预处理速度。

可选的，所述将所述待处理图像转化为三通道YUV图像，包括：

将所述待处理图像划分为第一Y通道图像和UV通道图像；

对所述第一Y通道图像进行调整，得到维度为(N,1,W,H)的Y通道图像；

对所述UV通道图像进行分离，得到维度为(N,1,W*0.5,H*0.5)的U通道图像和维度为(N,1,W*0.5,H*0.5)的V通道图像；

分别对所述维度为(N,1,W*0.5,H*0.5)的U通道图像和所述维度为(N,1,W*0.5,H*0.5)的V通道图像进行2倍采样，得到维度为(N,1,W,H)的U通道图像和维度为(N,1,W,H)的V通道图像；

将维度为(N,1,W,H)的Y通道图像、U通道图像和V通道图像进行合并，得到维度为(N,3,W,H)的YUV图像；

其中，所述N表示所述待处理图像的个数，所述1表示所述Y通道图像、所述U通道图像或所述V通道图像为1个通道，所述3表示所述YUV图像为3个通道，所述W等于所述待处理图像的宽度，所述H等于所述待处理图像的高度。

这样，借助上述分离、采样和合并的过程，可以高效地获得满足后续卷积运算需求的三通道YUV图像。

可选的，所述对所述UV通道图像进行分离，得到维度为(N,1,W*0.5,H*0.5)的U通道图像和维度为(N,1,W*0.5,H*0.5)的V通道图像，包括：

将所述UV通道图像调整为维度为(N,2,W*H*0.25)的UV通道图像；

对所述维度为(N,2,W*H*0.25)的UV通道图像进行分离，得到维度为(N,1,W*H*0.25)的U通道图像和维度为(N,1,W*H*0.25)的V通道图像；