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

权利要求书

1.一种图像处理方法，其特征在于，应用于电子设备，所述电子设备包括人工智能AI芯片，所述方法包括：

基于所述AI芯片对应的高速内存的内存大小，将待处理图像分为多块子图像，所述子图像所占内存小于所述内存大小；

将每块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的子图像进行指定图像处理；

对进行所述指定图像处理后的多块子图像进行合成，得到合成图像。

2.根据权利要求1所述的方法，其特征在于，所述基于所述AI芯片对应的高速内存的内存大小，将所述待处理图像分为多块子图像，包括：

基于所述AI芯片对应的高速内存的内存大小以及所述待处理图像所占内存，确定对所述待处理图像进行分割的分割数量；

将所述待处理图像分为所述分割数量对应的多块子图像。

3.根据权利要求2所述的方法，其特征在于，所述基于所述AI芯片对应的高速内存的内存大小以及所述待处理图像所占内存，确定对所述待处理图像进行分割的分割数量，包括：

若所述指定图像处理包括像素填充，确定所述像素填充对应的填充像素所占内存；

基于所述AI芯片对应的高速内存的内存大小、所述待处理图像所占内存以及所述像素填充对应的填充像素所占内存，确定对所述待处理图像进行分割的分割数量，其中，以所述分割数量分割获得的子图像所占内存与所述填充像素所占内存的总和小于所述内存大小。

4.根据权利要求1所述的方法，其特征在于，所述将每块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的子图像进行指定图像处理，包括：

每次将至少两块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的每块子图像并行进行指定图像处理；

重复所述每次将至少两块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的每块子图像并行进行指定图像处理，直至对所述多块子图像中每块子图像完成所述指定图像处理。

5.根据权利要求4所述的方法，其特征在于，在所述每次将至少两块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的每块子图像并行进行指定图像处理之前，所述方法还包括：

基于所述AI芯片的芯片核心的数量，创建至少两个线程，其中，所述至少两个线程的数量与所述芯片核心的数量对应；

所述每次将至少两块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的每块子图像并行进行指定图像处理，包括：

利用所述至少两个线程，每次将至少两块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的每块子图像并行进行指定图像处理，其中，每个线程对应一个芯片核心。

6.根据权利要求4所述的方法，其特征在于，在所述每次将至少两块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的每块子图像并行进行指定图像处理之前，所述方法还包括：

判断所述AI芯片的剩余资源量是否大于资源阈值；

若所述剩余资源量大于资源阈值，执行所述每次将至少两块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的每块子图像并行进行指定图像处理。

7.根据权利要求6所述的方法，其特征在于，在所述判断所述AI的剩余资源量是否大于资源阈值之后，所述方法还包括：

若所述剩余资源量小于或等于所述资源阈值，每次将单块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的子图像并行进行指定图像处理；

重复所述每次将单块子图像对应的图像数据依次写入所述高速内存，并基于所述AI芯片对所述高速内存中写入的子图像并行进行指定图像处理的步骤，直至对所述多块子图像中每块子图像完成所述指定图像处理。