[发明专利]车厢三维图像生成方法及装置

权利要求书

1.一种车厢三维图像生成方法，其特征在于，包括：

获取车厢图像集合；

判断所述车厢图像集合中的车厢图像是否包含完整的车厢；

对包含完整车厢的所述车厢图像进行预处理；

将处理后的所述车厢图像输入到预先建立的车厢分类模型及生成对抗网络模型，分别输出所述车厢图像对应的车型及车厢内壁估测结果，所述车厢内壁估测结果包括所述车厢图像的内壁的四个顶点的像素坐标值；

判断所述车型及所述车厢内壁估测结果是否在误差范围内，如在，融合处理所述车厢图像和所述车厢内壁估测结果；

将所述融合处理后的结果输入到生成模型，生成所述车厢的三维图像；

其中，所述判断所述车型及所述车厢内壁估测结果是否在误差范围内，如在，融合处理所述车厢图像和所述车厢内壁估测结果，包括：

根据所述车型确定所述车厢图像对应的车厢的标准尺寸；

根据所述车厢图像的内壁的四个顶点的像素坐标值确定所述车厢图像对应的车厢的预测尺寸；

比较所述标准尺寸与所述预测尺寸，若所述标准尺寸与所述预测尺寸在所述误差范围内，则获取所述分类模型输出的所述车厢图像的车型集合；

根据所述车型集合的众数和/或置信度指标确定所述分类模型输出的最终车型；

基于所述最终车型确定所述最终车型对应的标准车厢尺寸；

基于所述车厢图像的内壁的四个顶点的像素坐标值按照所述标准车厢尺寸估测车框的四个顶点的像素坐标值。

2.根据权利要求1所述的车厢三维图像生成方法，建立所述生成对抗网络模型包括：

获取训练样本集合，所述训练样本集合包括至少一帧空车厢图像；

对所述空车厢图像进行预处理；

对处理后的所述空车厢图像进行训练，得到生成对抗网络模型。

3.根据权利要求2所述的车厢三维图像生成方法，其特征在于，对所述车厢图像或空车厢图像进行预处理包括：

对所述车厢图像或空车厢图像的车框进行延伸处理；

将所述车厢图像或空车厢图像的中央区域进行蒙版处理。

4.根据权利要求1所述的车厢三维图像生成方法，其特征在于，在获取车厢图像集合之前，该方法包括：

获取车辆图像帧序列；

计算相邻帧之间的相似度；

若所述相似度小于等于第一阈值，则获取所述相邻帧的后一帧作为车厢图像；

对所述车厢图像进行在线数据增广处理，得到所述车厢图像集合。

5.根据权利要求1所述的车厢三维图像生成方法，其特征在于，将所述融合处理后的结果输入到生成模型，生成所述车厢的三维图像，包括：

利用所述最终车型和所述车厢的车框对应的四个顶点的像素坐标值以及所述车厢内壁估测结果进行多视图几何变换，建立所述车厢的三维图像。