[发明专利]图像裁剪方法、装置、设备及存储介质

说明书

本公开涉及一种图像裁剪方法、装置、设备及存储介质。获取待裁剪图像；基于预先训练好的图像检测模型，确定每个子图像的核心预测图、每个子图像的距离预测图以及每个子图像的角度预测图；针对每个子图像，基于核心预测图中像素点的概率值，确定每个子图像的核心像素集合；对于每个子图像的核心像素集合中的每个核心像素点，确定其对应于距离预测图中的至少一个距离值以及其对应于角度预测图中的至少一个角度值；基于每个子图像中每个核心像素点的坐标、每个核心像素点对应的至少一个距离值和每个核心像素点对应的至少一个角度值，确定每个子图像的预测页面框；基于预测页面框对待裁剪图像进行剪裁，得到目标剪裁图像。这样提高了图像的裁剪效果。

技术领域

本公开涉及图像处理技术领域，尤其涉及一种图像裁剪方法、装置、设备及存储介质。

背景技术

随着多媒体时代的快速发展，以及拍照的方便化和多样化，用户对图像的裁剪需求越来越多。

然而，当图像的复杂度较高时，从一张图像中裁剪出至少一个子图像的难度较大，裁剪效果不佳。

发明内容

为了解决上述技术问题，本公开提供了一种图像裁剪方法、装置、设备及存储介质。

第一方面，本公开提供了一种图像裁剪方法，该方法包括：

获取待裁剪图像，其中，所述待裁剪图像包括至少一个子图像；

基于预先训练好的图像检测模型，确定每个子图像的核心预测图、每个子图像的距离预测图以及每个子图像的角度预测图；

针对所述每个子图像，基于所述核心预测图中像素点的概率值，确定所述每个子图像的核心像素集合；

对于所述每个子图像的核心像素集合中的每个核心像素点，确定其对应于所述距离预测图中的至少一个距离值以及其对应于所述角度预测图中的至少一个角度值；

基于所述每个子图像中所述每个核心像素点的坐标、所述每个核心像素点对应的至少一个距离值和每个所述核心像素点对应的至少一个角度值，确定所述每个子图像的预测页面框；

基于所述预测页面框对所述待裁剪图像进行剪裁，得到目标剪裁图像。

第二方面，本公开提供了一种图像裁剪装置，该装置包括：

待裁剪图像模块，用于获取待裁剪图像，其中，所述待裁剪图像包括至少一个子图像；

图像检测模块，用于基于预先训练好的图像检测模型，确定每个子图像的核心预测图、每个子图像的距离预测图以及每个子图像的角度预测图；

第一确定模块，用于针对所述每个子图像，基于所述核心预测图中像素点的概率值，确定所述每个子图像的核心像素集合；

第二确定模块，用于对于所述每个子图像的核心像素集合中的每个核心像素点，确定其对应于所述距离预测图中的至少一个距离值以及其对应于所述角度预测图中的至少一个角度值；

第三确定模块，用于基于所述每个子图像中所述每个核心像素点的坐标、所述每个核心像素点对应的至少一个距离值和每个所述核心像素点对应的至少一个角度值，确定所述每个子图像的预测页面框；

图像裁剪模块，用于基于所述预测页面框对所述待裁剪图像进行剪裁，得到目标剪裁图像。

第三方面，本公开实施例还提供了一种图像裁剪设备，该设备包括：

一个或多个处理器；

存储装置，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现第一方面所提供的图像裁剪方法。