[发明专利]回归视频画面中屏幕的方法、装置、可读介质及电子设备

权利要求书

1.一种回归视频画面中屏幕的方法，其特征在于，所述方法包括：

针对视频的一帧图像，将所述图像输入预设网络回归模型，得到所述预设网络回归模型输出的图像的分类结果，其中，所述分类结果为含有屏幕或不含屏幕，所述预设网络回归模型的边框BBox回归用于，在分类结果为含有屏幕时，回归输出按固定顺序依次表征屏幕的四个角的坐标；

响应于获取到用于表征屏幕的四个角的坐标，根据所述四个角的坐标，回归所述屏幕的边框；

根据所述预设网络回归模型输出的所述屏幕的四个角的坐标，计算所述屏幕的左上角到左下角的距离L_l和所述屏幕的右上角到右下角的距离L_r；

根据所述屏幕的四个角的坐标、L_l和L_r，计算所述屏幕左上角与左下角连线构成的边到右上角与右下角连线构成的边之间的距离L₀；

确定真实世界中屏幕的高宽比R；

根据L₀和R，通过公式确定所述屏幕相对于拍摄所述视频的摄像头的角度偏差；

其中，

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

根据所述预设网络回归模型输出的所述屏幕的四个角的坐标，计算所述屏幕的面积占所述图像面积的比例。

3.根据权利要求1或2所述的方法，其特征在于，所述预设网络回归模型是通过以下方式训练得到的：

针对图像集，根据用户输入的标注指令对该图像集内包含有屏幕的图像进行标注，所述标注指令为根据固定顺序依次表示的该屏幕的四个角的坐标；

根据所述图像集和损失函数训练，得到所述预设网络回归模型；

其中，所述损失函数中的定位损失函数为Smooth L1，Smooth L1的输入和输出为根据固定顺序依次表征屏幕的四个角的坐标。

4.根据权利要求3所述的方法，其特征在于，根据用户输入的标注指令对该图像集内包含有屏幕的图像进行标注，包括：

在所述含有屏幕的图像的屏幕的至少一个角在所述图像外时，根据用户输入的标注指令将所述屏幕在所述图像外的角标注在所述图像外。

5.根据权利要求3所述的方法，其特征在于，所述损失函数中的置信度损失函数为交叉熵损失函数。

6.一种回归视频画面中屏幕的装置，其特征在于，包括：

计算模块，用于针对视频的一帧图像，将所述图像输入预设网络回归模型，得到所述预设网络回归模型输出的图像的分类结果，其中，所述分类结果为含有屏幕或不含屏幕，所述预设网络回归模型的边框BBox回归用于，在分类结果为含有屏幕时，回归输出按固定顺序依次表征屏幕的四个角的坐标；

回归模块，用于响应于获取到用于表征屏幕的四个角的坐标，根据所述四个角的坐标，回归所述屏幕的边框；

角度偏差计算模块，用于根据所述预设网络回归模型输出的所述屏幕的四个角的坐标，计算所述屏幕的左上角到左下角的距离L_l和所述屏幕的右上角到右下角的距离L_r；根据所述屏幕的四个角的坐标、L_l和L_r，计算所述屏幕左上角与左下角连线构成的边到右上角与右下角连线构成的边之间的距离L₀；确定真实世界中屏幕的高宽比R；根据L₀和R，通过公式确定所述屏幕相对于拍摄所述视频的摄像头的角度偏差；

其中，

7.根据权利要求6所述的装置，其特征在于，所述装置还包括：

面积占比模块，用于根据所述预设网络回归模型输出的所述屏幕的四个角的坐标，计算所述屏幕的面积占所述图像面积的比例。

8.一种计算机可读介质，其上存储有计算机程序，其特征在于，该程序被处理装置执行时实现权利要求1-5中任一项所述方法的步骤。

9.一种电子设备，其特征在于，包括：

存储装置，其上存储有计算机程序；

处理装置，用于执行所述存储装置中的所述计算机程序，以实现权利要求1-5中任一项所述方法的步骤。