[发明专利]违法车辆检测方法、装置、计算机设备和存储介质

权利要求书

1.一种违法车辆检测方法，其特征在于，所述方法包括：

获取待检测图片；

将所述待检测图片输入至语义分割模型得到分割结果；

当所述分割结果中包括牵引绳和且所述牵引绳两端均存在识别车辆，则根据所述识别车辆的行驶方向，获取位于所述牵引绳前端的目标车辆；

判断所述目标车辆是否为预设类型的车辆；

当所述目标车辆为预设类型的车辆，输出所述目标车辆违法的结果。

2.根据权利要求1所述的方法，其特征在于，所述将所述待检测图片输入至语义分割模型得到分割结果之后，还包括：

当所述分割结果中不包括牵引绳，则输出所述待检测图片中不存在违法车辆的结果。

3.根据权利要求2所述的方法，其特征在于，所述将所述待检测图片输入至语义分割模型得到分割结果之后，还包括：

当所述分割结果中包括牵引绳，且所述牵引绳两端中至少一个不存在识别车辆，则输出所述待检测图片中不存在违法车辆的结果。

4.根据权利要求1至3任意一项所述的方法，其特征在于，所述将所述待检测图片输入至语义分割模型得到分割结果，包括：

获取所述待检测图片中的当前特征点，并获取与所述当前特征点形成十字交叉关系的待判断特征点；

计算所述当前特征点和所述待判断特征点属于同一语义的参考值；

当所述参考值大于预设值时，则所述当前特征点和所述待判断特征点属于同一语义，否则，所述当前特征点和所述待判断特征点不属于同一语义；

继续获取所述当前特征点的下一特征点，将所述下一特征点作为当前特征点，并继续获取与所述当前特征点形成十字交叉关系的待判断特征点，直至所述待检测图片中的特征点均判断完成；

获取属于同一语义的特征点，并根据属于同一语义的特征点得到分割结果。

5.根据权利要求1至3任意一项所述的方法，其特征在于，所述根据所述识别车辆的行驶方向，获取位于所述牵引绳前端的目标车辆之前，还包括：

确定所述识别车辆的车头位置和/或车尾位置；

根据所述车头位置和/或车尾位置得到所述识别车辆的行驶方向。

6.根据权利要求1至3任意一项所述的方法，其特征在于，所述判断所述目标车辆是否为预设类型的车辆，包括：

获取预设分类模型，并确定所述预设分类模型中各个通道对应的层数；

将所述层数减少至预设层；

根据层数减少至预设层的各个通道得到改进后的分类模型；

将所述目标车辆输入至改进后的分类模型，以判断所述目标车辆是否为预设类型的车辆。

7.一种违法车辆检测装置，其特征在于，所述装置包括：

图片获取模块，用于获取待检测图片；

语义分割模块，用于将所述待检测图片输入至语义分割模型得到分割结果；

目标车辆获取模块，用于当所述分割结果中包括牵引绳和且所述牵引绳两端均存在识别车辆，则根据所述识别车辆的行驶方向，获取位于所述牵引绳前端的目标车辆；

车辆类型判断模块，用于判断所述目标车辆是否为预设类型的车辆；

输出模块，用于当所述目标车辆为预设类型的车辆，输出所述目标车辆违法的结果。

8.根据权利要求7所述的装置，其特征在于，所述输出模块还用于当所述分割结果中不包括牵引绳，则输出所述待检测图片中不存在违法车辆的结果。

9.一种计算机设备，包括存储器和处理器，所述存储器存储有计算机程序，其特征在于，所述处理器执行所述计算机程序时实现权利要求1至6中任一项所述的方法的步骤。

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