[发明专利]一种车轴计数方法、装置、机器可读介质及设备

权利要求书

1.一种车轴计数方法，其特征在于，包括：

通过第一检测模型对待检测图片进行目标检测，到得一个或多个车辆框；

通过第二检测模型对所述一个或多个车辆框进行目标检测，得到多个车轮框；

根据所述多个车轮框得到车轴的数量。

2.根据权利要求1所述的车轴计数方法，其特征在于，所述车轮框的数据包括车轮框的数量以及车轮框的类型。

3.根据权利要求2所述的车轴计数方法，其特征在于，根据所述车轮框的长宽比确定所述车轮框的类型。

4.根据权利要求2或3所述的车轴计数方法，其特征在于，所述车轮框的类型包括：一轴车轮框、多轴车轮框。

5.根据权利要求1所述的车轴计数方法，其特征在于，若通过第一检测模型对待检测图片进行目标检测得到多个车辆框，则对所述多个车辆框进行筛选，得到目标车辆框。

6.根据权利要求5所述的车轴计数方法，其特征在于，所述对多个车辆框进行筛选，包括：

获取多个车辆框的面积；

从多个车辆框中筛选出区域面积最大的车辆框作为目标车辆框。

7.根据权利要求1所述的车轴计数方法，其特征在在于，所述通过第一检测模型对待检测图片进行目标检测，包括：

将待检测图片输入到第一检测模型，得到所述待检测图片中的一个或多个车辆框。

8.根据权利要求1或7所述的车轴计数方法，其特征在于，所述通过第二检测模型对所述一个或多个车辆框进行目标检测，包括：

将所述一个或多个车辆框框住的区域图像输入到第二检测模型，得到车辆框中的多个车轮框。

9.一种车轴计数装置，其特征在于，包括：

车辆检测模块，用于通过第一检测模型对待检测图片进行目标检测，到得一个或多个车辆框；

车轮检测模块，用于通过第二检测模型对所述一个或多个车辆框进行目标检测，得到多个车轮框；

车轴检测模块，用于根据所述多个车轮框得到车轴的数量。

10.根据权利要求8所述的车轴计数装置，其特征在于，所述车轮框的数据包括车轮框的数量以及车轮框的类型。

11.根据权利要求10所述的车轴计数装置，其特征在于，根据所述车轮框的长宽比确定所述车轮框的类型。

12.根据权利要求10或11所述的车轴计数装置，其特征在于，所述车轮框的类型包括：一轴车轮框、多轴车轮框。

13.根据权利要求9所述的车轴计数装置，其特征在于，该装置还包括筛选模块，用于在通过第一检测模型对待检测图片进行目标检测得到多个车辆框时，对所述多个车辆框进行筛选，得到目标车辆框。

14.根据权利要求13所述的车轴计数装置，其特征在于，所述筛选模块包括：

面积获取子模块，用于获取多个车辆框的面积；

筛选子模块，用于从多个车辆框中筛选出区域面积最大的车辆框作为目标车辆框。

15.一种车轴计数设备，其特征在于，包括：

一个或多个处理器；和

其上存储有指令的一个或多个机器可读介质，当所述一个或多个处理器执行时，使得所述设备执行如权利要求1-8中一个或多个所述的方法。

16.一个或多个机器可读介质，其特征在于，其上存储有指令，当由一个或多个处理器执行时，使得设备执行如权利要求1-8中一个或多个所述的方法。