一种图像数据的插值方法、装置、存储介质及电子设备与流程

本技术涉及计算机，尤其涉及一种图像数据的插值方法、装置、存储介质及电子设备。

背景技术：

1、目前，应用图像传感器的场景越来越多。传统的拜耳型(bayer)图像传感器，由红蓝绿(rgb)三色单元阵列组成，每种颜色的单元用于感应不同波段的可见光，得到不同颜色的图像数据，根据三色图像数据生成彩色图像。

2、现有技术中，为了扩展图像传感器的应用场景，通过在三色单元阵列中添加感应红外波段光的红外(infrared radiation，ir)单元，构成彩色-红外(rgb-ir)传感器，实现同时感应可见光以及红外光。进一步扩展了图像传感器的应用场景。

3、但是，由于目前rgb-ir传感器通常是将传感器阵列中的绿光单元(即，g单元)替换为ir单元，导致后续进行图像图处理的图像信号处理单元(image signal processor，isp)需要重新开发，使应用rgb-ir传感器成本增加。基于此，本技术提供一种图像数据的插值方法。

技术实现思路

1、本技术提供一种图像数据的插值方法、装置、存储介质及电子设备，以部分的解决现有技术存在的上述问题。

2、本技术采用下述技术方案：

3、本技术提供了一种图像数据的插值方法，包括：

4、获取彩色-红外图像传感器采集的图像数据，确定所述图像数据中，由各三色单元以及各红外单元分别采集的图像数据；

5、根据所述图像传感器中各单元的位置关系，对于至少部分红外单元，确定所述红外单元对角线上相邻的其他红外单元以及相邻的三色单元，作为相邻单元；

6、根据所述红外单元以及所述相邻单元采集的图像数据，确定在采集所述图像数据时，所述红外单元处的绿色光的分量插值，以及图像数据的梯度值的大小关系；

7、根据所述大小关系以及所述分量插值，确定在所述红外单元处插值的绿色图像数据；

8、根据插值得到的图像数据以及所述图像数据中各三色单元分别采集的图像数据，确定用于生成彩色图像的图像数据。

9、可选地，所述彩色-红外图像传感器中红外单元与红色单元和蓝色单元邻接，对角线方向与绿色单元相邻。

10、可选地，确定所述红外单元对角线上相邻的其他红外单元，作为相邻单元，包括：

11、确定所述红外单元主对角线上相隔一个绿色单元的其他红外单元，为相邻单元；

12、确定所述红外单元副对角线上相隔一个绿色单元的其他红外单元，为相邻单元。

13、可选地，根据所述红外单元以及所述相邻单元采集的图像数据，确定在采集所述图像数据时，所述红外单元处的绿色光的分量插值，包括：

14、根据所述红外单元主对角线上绿光单元的图像数据的差值、所述红外单元与主对角线上其他红外单元的图像数据的差值以及所述红外单元的图像数据，确定主对角线的绿色光的分量插值；

15、根据所述红外单元副对角线上绿光单元的图像数据的差值、所述红外单元与副对角线上其他红外单元的图像数据的差值以及所述红外单元的图像数据，确定副对角线的绿色光的分量插值。

16、可选地，根据所述红外单元以及所述相邻单元采集的图像数据，确定在采集所述图像数据时，所述图像数据的梯度值的大小关系，包括：

17、根据所述红外单元主对角线上绿光单元的差值，确定主一阶梯度；

18、根据所述红外单元分别与主对角线上相邻的其他红外单元的图像数据的差值，确定主二阶梯度；

19、根据所述红外单元副对角线上绿光单元的差值，确定副一阶梯度；

20、根据所述红外单元分别与副对角线上相邻的其他红外单元的图像数据的差值，确定副二阶梯度；

21、根据所述主一阶梯度以及所述主二阶梯度，确定主对角线梯度，根据所述副一阶梯度以及所述副二阶梯度，确定副对角线梯度；

22、确定所述主对角线梯度与所述副对角线梯度的大小关系。

23、可选地，根据所述大小关系以及所述分量插值，确定在所述红外单元处插值的绿色图像数据，包括：

24、当所述主对角线梯度大于所述副对角线梯度，则根据所述副对角线的绿色光的分量插，确定在所述红外单元处插值的绿色图像数据；

25、当所述主对角线梯度小于所述副对角线梯度，则根据所述主对角线的绿色光的分量插，确定在所述红外单元处插值的绿色图像数据；

26、当所述主对角线梯度等于所述副对角线梯度，则根据所述主对角线的绿色光的分量插以及所述副对角线的绿色光的分量插的均值，确定在所述红外单元处插值的绿色图像数据。

27、可选地，确定所述主对角线梯度与所述副对角线梯度的大小关系，包括：

28、根据预设的冗余参数，判断所述主对角线梯度与所述副对角线梯度的差值是否小于所述冗余参数；

29、若是，则确定所述主对角线梯度等于所述副对角线梯度；

30、若否，则根据所述主对角线梯度等于所述副对角线梯度，确定所述主对角线梯度与所述副对角线梯度的大小关系。

31、可选地，根据插值得到的图像数据以及所述图像数据中各三色单元分别采集的图像数据，确定用于生成彩色图像的图像数据，包括：

32、根据所述图像传感器中所述红外单元的位置，将所述图像数据中所述红外单元的图像数据替换为插值的绿色图像数据，确定用于生成彩色图像的图像数据；

33、以及根据各红外单元的图像数据，确定用于生成红外图像的图像数据。

34、本技术提供了一种图像数据的插值装置，

35、所述装置应用于彩色-红外图像传感器，包括：

36、获取模块，获取所述彩色-红外图像传感器采集的图像数据，确定所述图像数据中，由各三色单元以及各红外单元分别采集的图像数据；

37、定位模块，根据所述图像传感器中各单元的位置关系，对于至少部分红外单元，确定所述红外单元对角线上相邻的其他红外单元以及相邻的三色单元，作为相邻单元；

38、梯度确定模块，根据所述红外单元以及所述相邻单元采集的图像数据，确定在采集所述图像数据时，所述红外单元处的绿色光的分量插值，以及图像数据的梯度值的大小关系；

39、插值确定模块，根据所述大小关系以及所述分量插值，确定在所述红外单元处插值的绿色图像数据；

40、数据组合模块，根据插值得到的图像数据以及所述图像数据中各三色单元分别采集的图像数据，确定用于生成彩色图像的图像数据。

41、本技术提供了一种计算机可读存储介质，所述存储介质存储有计算机程序，所述计算机程序被处理器执行时实现上述图像数据的插值的方法。

42、本技术提供了一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现上述图像数据的插值的方法。

43、本技术采用的上述至少一个技术方案能够达到以下有益效果：

44、首先确定图像数据中由三色单元和红外单元分别采集的图像数据，之后根据图像传感器中各单元的位置关系，对于至少部分红外单元，确定红外单元对角线上相邻的其他红外单元以及相邻的三色单元，作为该红外单元的相邻单元，根据该红外单元以及该相邻单元采集的图像数据，确定该红外单元处两种对角线方向的绿色光的分量插值，以及在两种对角线上图像数据的梯度值的大小关系，识别该红外单元处是否存在有对角线方向上的色彩梯度大幅变化的情况，并识别结果确定基于哪些相邻单元的图像数据恢复该红外单元处的绿色图像数据，可以更准确的进行绿色差值，得到生成彩色图像的图像数据。

45、由于在进行差值时，考虑了红外单元相邻单元的颜色变化情况，使得可以针对性的确定采用哪些图像数据进行确定插值结果，还原得到更准确的图像数据。