[发明专利]一种车身广告植入方法、装置、电子设备及存储介质

说明书

本申请提供一种车身广告植入方法、装置、电子设备及存储介质。该方法包括：获取待处理视频对应的多帧待处理图像；针对每帧待处理图像进行车体检测，获得车辆的轨迹图像集；其中，轨迹图像集包括连续的多帧待处理图像，且连续的多帧待处理图像中均包括同一车辆；针对每辆车的轨迹图像集中的多帧待处理图像进行语义分割，获得目标区域；获取广告图像，将广告图像植入待处理图像对应的目标区域中。本申请实施例通过对待处理图像进行车体检测，自动检测出包含车辆的图像，并对包含车辆的图像进行语义分割获得目标区域，最后将广告图像植入目标区域中，全过程自动化实现，无需人工植入，提高了对广告植入的效率。

技术领域

本申请涉及图像处理技术领域，具体而言，涉及一种车身广告植入方法、装置、电子设备及存储介质。

背景技术

都市剧、短视频中，车辆是频繁出现的物体，而车辆本身又是户外广告极佳的途径，具有很强的现代感和视觉冲击力。为了起到宣传效果，往往会在视频中车辆的车身上植入广告。

现有技术中，在车身上植入广告的方法是通过人工对视频中的每帧画面进行判断，判断该画面上是否有车辆，并在有车辆的画面中将广告植入车身上，其植入广告的效率很低。

发明内容

本申请实施例的目的在于提供一种车身广告植入方法、装置、电子设备及存储介质，用以提高对视频中的车身植入广告的效率。

第一方面，本申请实施例提供一种车身广告植入方法，包括：获取待处理视频对应的多帧待处理图像；针对每帧待处理图像进行车体检测，获得车辆的轨迹图像集；其中，所述轨迹图像集包括连续的多帧待处理图像，且连续的多帧待处理图像中均包括同一车辆；针对每辆车的轨迹图像集中的多帧待处理图像进行语义分割，获得目标区域；获取广告图像，将所述广告图像植入待处理图像对应的目标区域中。

本申请实施例通过对待处理图像进行车体检测，自动检测出包含车辆的图像，并对包含车辆的图像进行语义分割获得目标区域，最后将广告图像植入目标区域中，全过程自动化实现，无需人工植入，提高了对广告植入的效率。

进一步地，所述针对每帧待处理图像进行车体检测，获得车辆的轨迹图像集，包括：利用车体检测模型对每帧待处理图像进行车体检测，获得包含车辆的待处理图像中的车体信息；将相邻两帧包含车辆的待处理图像的车体信息进行匹配，获得车辆的轨迹图像集。本申请实施例通过比较相邻两帧待处理图像的车体信息判断待处理图像中的车辆是否为同一辆，从而可以获得属于同一辆车的轨迹图像集，以便于后续对同一辆车的车体植入广告，避免了待处理视频内容杂乱。

进一步地，所述将相邻两帧包含车辆的待处理图像的车体信息进行匹配，获得车辆的轨迹图像集，包括：计算相邻两帧待处理图像的车体信息之间的交并比，若所述交并比大于预设阈值，则相邻两帧待处理图像中的车辆为同一车辆，将属于同一车辆的连续多帧待处理图像作为所述车辆的轨迹图像集。本申请实施例通过计算相邻两帧待处理图像中的车体信息的交并比能够准确的确定这两帧待处理图像中的车是否是同一辆。

进一步地，在利用车体检测模型对每帧待处理图像进行车体检测之前，所述方法还包括：获取第一训练样本集，所述第一训练样本集包括多张第一训练图像以及每张第一训练图像对应的第一标签；其中，所述第一训练图像包括至少一辆车，所述第一标签包括车辆的轮廓；将所述第一训练图像输入车体检测模型中，所述车体检测模型输出对应的第一预测结果；根据所述第一预测结果和对应的第一训练图像的第一标签对所述车体检测模型中的参数进行优化，获得训练好的车体检测模型。本申请实施例通过对车体检测模型进行训练，获得的训练好的车体检测模型能够准确识别出待处理图像中是否包含车辆。