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

摘要

本申请公开了一种基于图像的目标检测方法。包括：获取第一待检测图像；获取第一待检测图像所对应的第二待检测图像，第二待检测图像属于第二领域，通过跨领域图像检测模型获取第二待检测图像所对应的检测结果，检测结果包括目标对象的目标定位信息和目标类别信息，跨领域图像检测模型包括第一网络模型以及第二网络模型，第一网络模型用于将图像从第一领域转换为第二领域，第二网络模型用于对第二领域的图像进行区域定位。本申请还公开了一种模型训练的方法及装置。本申请通过跨领域图像检测模型，将待检测图像的风格转换为测试集图像的风格，进而对转换后的图像进行目标定位，从而提升图像定位的准确性，增强模型预测效果。

主权项

1.一种模型训练的方法，其特征在于，包括：获取待训练图像以及待测试图像，其中，所述待训练图像属于第一领域，所述待测试图像属于第二领域，所述第一领域与所述第二领域为不同的领域；通过第一待训练网络模型获取所述待训练图像所对应的测试风格图像，通过所述第一待训练网络模型获取所述待测试图像所对应的训练风格图像，其中，所述测试风格图像属于所述第二领域，所述训练风格图像属于所述第一领域；通过第二待训练网络模型获取候选区域所对应的预测概率值，其中，所述候选区域是从所述待测试图像或所述测试风格图像中提取的候选框；通过所述第二待训练网络模型获取所述测试风格图像所对应的第一预测类别概率值、第一预测定位值、第二预测类别概率值以及第二预测定位值；根据所述预测概率值、所述测试风格图像、所述训练风格图像、所述第一预测类别概率值、所述第一预测定位值、所述第二预测类别概率值以及所述第二预测定位值，采用目标损失函数对所述第一待训练网络模型以及所述第二待训练网络模型进行训练，得到所述跨领域图像检测模型。