一种模型训练方法、跨域目标检测方法及电子设备技术

技术编号:41354366 阅读:17 留言:0更新日期:2024-05-20 10:06
本申请提供一种模型训练方法、跨域目标检测方法及电子设备,用于改善模型的检测准确率下降的问题。该方法包括:获取有标签的源域数据集和无标签的目标域数据集,目标域数据集包括:目标域图像;根据源域数据集计算出学生模型中的源域检测头网络的源域损失;通过教师模型对目标域图像进行目标检测,获得目标域伪标签,教师模型的权重参数是被学生模型的权重参数所赋值获得的;根据目标域伪标签和目标域图像计算出学生模型中的目标域检测头网络的目标域损失;根据源域数据集和目标域数据集计算出学生模型中的多尺度掩膜对抗性对齐模块的对齐损失;根据源域损失、目标域损失和对齐损失对学生模型进行训练,获得训练后的学生模型。

【技术实现步骤摘要】

本申请涉及深度学习、计算机视觉和目标检测的,具体而言,涉及一种模型训练方法、跨域目标检测方法及电子设备


技术介绍

1、在目前的跨域目标检测过程中,大都是使用权重共享检测头网络的目标检测模型来对源域数据样本和目标域数据样本进行分类的,尽管权重共享检测头旨在促进跨域特征的迁移和共享,但在实际应用过程中发现,由于源域数据具有真实标注信息,模型在联合优化时往往会过于依赖于源域数据样本的有监督训练过程,从而让模型更偏向源域数据样本的优化而忽视了目标域数据样本的优化,使得教师模型生成的伪标签的质量下降,最终导致模型的检测准确率下降。


技术实现思路

1、本申请实施例的目的在于提供一种模型训练方法、跨域目标检测方法及电子设备,用于改善模型的检测准确率下降的问题。

2、本申请实施例提供了一种模型训练方法,包括:获取有标签的源域数据集和无标签的目标域数据集,目标域数据集包括:目标域图像;根据源域数据集计算出学生模型中的源域检测头网络的源域损失;通过教师模型对目标域图像进行目标检测,获得目标域伪标签,教师模型的权重参本文档来自技高网...

【技术保护点】

1.一种模型训练方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,所述源域数据集包括:源域图像和源域标签;所述根据所述源域数据集计算出学生模型中的源域检测头网络的源域损失,包括:

3.根据权利要求2所述的方法,其特征在于,所述根据所述目标域伪标签和所述目标域图像计算出所述学生模型中的目标域检测头网络的目标域损失,包括:

4.根据权利要求3所述的方法,其特征在于,所述根据所述源域数据集和所述目标域数据集计算出所述学生模型中的多尺度掩膜对抗性对齐模块的对齐损失,包括:

5.根据权利要求4所述的方法,其特征在于,所述多尺度掩膜对...

【技术特征摘要】

1.一种模型训练方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,所述源域数据集包括:源域图像和源域标签;所述根据所述源域数据集计算出学生模型中的源域检测头网络的源域损失,包括:

3.根据权利要求2所述的方法,其特征在于,所述根据所述目标域伪标签和所述目标域图像计算出所述学生模型中的目标域检测头网络的目标域损失,包括:

4.根据权利要求3所述的方法,其特征在于,所述根据所述源域数据集和所述目标域数据集计算出所述学生模型中的多尺度掩膜对抗性对齐模块的对齐损失,包括:

5.根据权利要求4所述的方法,其特征在于,所述多尺度掩膜对抗性对齐模块包括:全卷积解码器;所述通过所述多尺度掩膜对抗性对齐模块对所述随机掩膜、所述通道变量和所述目标域多尺度特征图进行计算,包括:<...

【专利技术属性】
技术研发人员:邹昆关圣贤董帅
申请(专利权)人:电子科技大学中山学院
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1