面向半监督旋转目标检测的密集伪标签方法、装置和设备
申请号:CN202411079817
申请日期:2024-08-07
公开号:CN118608753B
公开日期:2024-10-22
类型:发明专利
摘要
本发明涉及一种面向半监督旋转目标检测的密集伪标签方法、装置和设备。方法包括:构建半监督旋转目标检测的预训练模型,预训练模型包括第一模型与第二模型;获取数据集,对数据集进行弱数据增强后,输入第一模型进行处理,输出密集伪标签;以及对数据集进行强数据增强后,输入第二模型进行训练,得到第二训练结果;训练过程中,通过第一模型中各尺度图像对第二模型中对应的各尺度图像进行约束,约束时,第一模型中各尺度图像一一对应约束第二模型中各尺度图像;根据密集伪标签与第二训练结果进行损失计算,更新预训练模型。本发明适用于旋转目标的检测,能够对任意方向、小尺寸、密集分布的旋转对象进行有效识别。
技术关键词
标签方法
预训练模型
数据
特征金字塔
图像
标签装置
动态更新
计算机设备
无监督
小尺寸
存储器
处理器
表达式
模块
指数
对象