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

权利要求书

1.一种图像处理方法，其特征在于，包括：

获取待处理图像；

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

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

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

2.根据权利要求1所述的方法，其特征在于，所述将所述待处理图像转化为三通道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等于所述待处理图像的高度。

3.根据权利要求2所述的方法，其特征在于，所述对所述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通道图像；

将所述维度为(N,1,W*H*0.25)的U通道图像调整为所述维度为(N,1,W*0.5,H*0.5)的U通道图像，和将所述维度为(N,1,W*H*0.25)的V通道图像调整为所述维度为(N,1,W*0.5,H*0.5)的V通道图像。

4.根据权利要求2所述的方法，其特征在于，所述分别对所述维度为(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通道图像，包括：

利用nearest插值函数，分别对所述维度为(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通道图像。

5.根据权利要求1所述的方法，其特征在于，所述对所述三通道RGB图像进行预处理，包括：

利用批归一化函数对所述三通道RGB图像进行预处理。

6.根据权利要求5所述的方法，其特征在于，所述批归一化函数为：

其中，所述output表示预处理后图像的像素值，所述input表示预处理前图像的像素值，所述scale表示缩放系数，所述bias表示偏置值，所述mean表示均值，所述variance表示方差，所述epsilon表示预设的除法数值稳定系数。

7.根据权利要求1所述的方法，其特征在于，所述得到三通道RGB图像之后，所述方法还包括：

对所述三通道RGB图像的尺寸进行调整；

所述对所述三通道RGB图像进行预处理，包括：

对调整尺寸后的三通道RGB图像进行预处理。