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

说明书

本发明公开了一种车轴计数方法，包括：通过第一检测模型对待检测图片进行目标检测，到得一个或多个车辆框；通过第二检测模型对所述一个或多个车辆框进行目标检测，得到多个车轮框；根据所述多个车轮框得到车轴的数量。本发明针对人工查看效率低下的问题，设计了一套通过目标检测获取车轴数量的方法，该方法可以做到实时检测，实时出结果，克服了人工统计时间长的问题，准确率高，杜绝了人眼视觉盲区的限制。

技术领域

本发明涉及人工智能领域，具体涉及一种车轴计数方法、装置、机器可读介质及设备。

背景技术

2020年1月1日起，全国高速公路计费方式，将由计重收费变为按轴收费。现有的收费是通过地磅结合人眼观察的结果进行计算，但单纯通过地磅和人眼存在效率低下的问题。

发明内容

鉴于以上所述现有技术的缺点，本发明的目的在于提供一种车轴计数方法、装置、机器可读介质及设备，用于解决现有技术存在的问题。

为实现上述目的及其他相关目的，本发明提供一种车轴计数方法，包括：

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

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

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

可选地，所述车轮框的数据包括车轮框的数量以及车轮框的类型。

可选地，根据所述车轮框的长宽比确定所述车轮框的类型。

可选地，所述车轮框的类型包括：一轴车轮框、多轴车轮框。

可选地，若通过第一检测模型对待检测图片进行目标检测得到多个车辆框，则对所述多个车辆框进行筛选，得到目标车辆框。

可选地，所述对多个车辆框进行筛选，包括：

获取多个车辆框的面积；

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

可选地，所述通过第一检测模型对待检测图片进行目标检测，包括：

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

可选地，所述通过第二检测模型对所述一个或多个车辆框进行目标检测，包括：

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

为实现上述目的及其他相关目的，本发明提供一种车轴计数装置，包括：

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

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

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

可选地，所述车轮框的数据包括车轮框的数量以及车轮框的类型。

可选地，可选地，根据所述车轮框的长宽比确定所述车轮框的类型。

可选地，所述车轮框的类型包括：一轴车轮框、多轴车轮框。

可选地，该装置还包括筛选模块，用于在通过第一检测模型对待检测图片进行目标检测得到多个车辆框时，对所述多个车辆框进行筛选，得到目标车辆框。

可选地，所述筛选模块包括：

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