[发明专利]判断主光源的方位的方法、介质、装置和计算设备

权利要求书

1.一种判断主光源的方位的方法，包括：

获取针对目标场景采集的第一场景图，其中所述目标场景中包含有主光源，所述主光源是所述目标场景中光照强度最强的光源；

对所述第一场景图进行检测，以确定所述目标场景中用于判断所述主光源的方位的参照物；

对所述第一场景图进行阴影分割，以确定所述参照物经所述主光源照射而产生的阴影；以及

根据确定出的所述参照物和所述阴影，判断所述主光源在所述目标场景中的方位，

其中，对所述第一场景图进行检测，以确定所述目标场景中用于判断所述主光源的方位的参照物包括：

将所述第一场景图输入参照物检测网络，以使所述参照物检测网络输出具有突出显示部分的第二场景图，其中所述突出显示包括矩形框，所述矩形框用于粗略表示所述参照物的轮廓或局部；以及

将所述参照物简化为所述矩形框来判断所述主光源的方位，

其中，在对所述第一场景图进行阴影分割的过程中，执行以下操作：

先将所述第一场景图中包含的所有阴影部分的像素值设置为1，并将所述第一场景图中包含的所有非阴影部分的像素值设置为0，再将所述所有阴影部分中出现在所述矩形框中的阴影部分的像素值重置为0；

根据确定出的所述参照物和所述阴影，判断所述主光源在所述目标场景中的方位包括：确定出所述参照物的第一中心线；确定出所述阴影的第二中心线；确定所述第一中心线与所述第二中心线的夹角；以及，根据所述夹角估算所述主光源在所述目标场景中的方位；

在参照物包括n个，其中，n为大于0的整数的情况下，通过以下方案中的任意一种来确定所述主光源的方位：

方案1：针对n个参照物，先分别利用n个参照物来确定主光源的方位，得到n个方位确定结果，再对这n个方位确定结果求平均值，并将所述平均值作为所述主光源的方位的最终确定结果；

方案2：针对n个参照物，先分别利用n个参照物来确定主光源的方位，得到n个方位确定结果，再对所述n个方位确定结果求均方根，并将所述均方根作为所述主光源的方位的最终确定结果；

方案3：针对n个参照物，先分别利用其中任意n-1个参照物来确定的主光源的方位，得到n-1个方位确定结果，再对n-1个方位确定结果求平均值，同时利用剩余的参照物来确定的主光源的方位，得到所述剩余的参照物对应的方位确定结果，最后计算n-1个方位确定结果的平均值与所述剩余的参照物对应的方位确定结果的误差；如果误差小于或等于预设值X，则所述平均值的结果用于作为所述主光源的方位的最终确定结果；如果误差大于预设值X，则所述平均值的结果不可用于主光源的方位的最终确定结果。

2.根据权利要求1所述的方法，其中，对所述第一场景图进行阴影分割，以确定所述参照物经所述主光源照射而产生的阴影包括：

将所述第一场景图输入阴影分割网络，以使所述阴影分割网络输出对应的阴影图像；

基于不同物体的形态，对所述阴影图像进行阴影处理，以使所述阴影图像中不同物体的阴影得以分离；以及

根据所述突出显示部分的位置信息，从分离出的阴影中找出属于所述参照物的阴影部分。

3.根据权利要求2所述的方法，其中，在所述阴影分割网络对所述第一场景图进行处理的过程中：

利用所述阴影分割网络对所述第一场景图进行逐像素预测；

将所述第一场景图中包含的所有阴影部分的像素值设置为1；以及

将所述第一场景图中包含的所有非阴影部分的像素值设置为0。

4.根据权利要求3所述的方法，其中，在所述阴影分割网络对所述第一场景图进行处理的过程中：

在将所述第一场景图中包含的所有阴影部分的像素值设置为1之后，检测所述所有阴影部分中是否包含出现在所述突出显示部分中的伪阴影部分；以及

若所述所有阴影部分中包含出现在所述突出显示部分中的所述伪阴影部分，则将所述伪阴影部分的像素值设置为0。