[发明专利]一种目标检测模型的训练方法及目标检测方法、装置

说明书

本公开提供了一种目标检测模型的训练方法及目标检测方法、装置，涉及人工智能中的图像识别、深度学习等技术领域。具体实现方案为：获取样本训练图像以及样本训练图像的标签信息；利用样本训练图像及样本训练图像的标签信息，对轻量化目标检测模型进行训练，得到预训练轻量化目标检测模型；在预训练轻量化目标检测模型中加入量化模块，得到初始量化目标检测模型；再利用样本训练图像及样本训练图像的标签信息，对初始量化目标检测模型进行训练，得到训练好的轻量化目标检测模型，提高了模型的训练速度。

技术领域

本公开涉及人工智能技术领域，尤其涉及图像识别、深度学习等技术领域，具体涉及一种目标检测模型的训练方法及目标检测方法、装置。

背景技术

通信塔也称信号发射塔或信号塔，是通信运营商、广播电视等部门架设信号发射天线或微波传输设备的基础设备，不同类别的通信塔运营和维护方式不同，使得通信塔类别的识别较为重要。

发明内容

本公开提供了一种目标检测模型的训练方法及目标检测方法、装置。

根据本公开的第一方面，提供了一种目标检测模型的训练方法，包括：

获取样本训练图像以及所述样本训练图像的标签信息；

将所述样本训练图像输入轻量化目标检测模型中进行信号塔的预测，得到所述样本训练图像的第一预测结果；

根据所述样本训练图像的第一预测结果以及所述样本训练图像的标签信息计算第一损失，并根据第一损失对所述轻量化目标检测模型的训练参数进行调整，得到预训练轻量化目标检测模型；

在所述预训练轻量化目标检测模型中加入量化模块，得到初始量化目标检测模型；

将所述样本训练图像输入所述初始量化目标检测模型中进行信号塔的预测，得到所述样本训练图像的第二预测结果；

根据所述样本训练图像的第二预测结果以及所述样本训练图像的标签信息计算第二损失，并根据第二损失对所述初始量化目标检测模型的训练参数进行调整，得到训练好的轻量化目标检测模型。

根据本公开的第二方面，提供了一种目标检测方法，包括：

获取待检测图像；

将所述待检测图像输入训练好的轻量化目标检测模型中进行信号塔的检测，得到所述待检测图像的检测结果；其中，所述训练好的轻量化目标检测模型为：采用第一方面所述的方法训练得到的，所述训练好的轻量化目标检测模型用于对所述待检测图像中的信号塔进行检测，所述检测结果包括：类别检测结果和位置检测结果。

根据本公开的第三方面，提供了一种目标检测模型的训练装置，包括：

样本获取模块，用于获取样本训练图像以及所述样本训练图像的标签信息；

第一预测模块，用于将所述样本训练图像输入轻量化目标检测模型中进行信号塔的预测，得到所述样本训练图像的第一预测结果；

第一训练模块，用于根据所述样本训练图像的第一预测结果以及所述样本训练图像的标签信息计算第一损失，并根据第一损失对所述轻量化目标检测模型的训练参数进行调整，得到预训练轻量化目标检测模型；

模型量化模块，用于在所述预训练轻量化目标检测模型中加入量化模块，得到初始量化目标检测模型；

第二预测模块，用于将所述样本训练图像输入所述初始量化目标检测模型中进行信号塔的预测，得到所述样本训练图像的第二预测结果；

第二训练模块，用于根据所述样本训练图像的第二预测结果以及所述样本训练图像的标签信息计算第二损失，并根据第二损失对所述初始量化目标检测模型的训练参数进行调整，得到训练好的轻量化目标检测模型。