一种适用于电力场景目标检测的预训练方法及系统
摘要:
本发明公开了一种适用于电力场景目标检测的预训练方法及系统,包括:获取清晰电力场景图像数据集,所述清晰电力场景图像数据集无需人工标注;在对无标签的图像进行不同的预处理后,将预处理后的图像输入到多个编码器中,设计多种损失函数对编码器的输出进行运算,使编码器能从大量无标签的电力场景图像中提取特征,最终得到用于电力场景的目标检测任务的预训练模型;使用有标签的电力场景目标检测数据集对预训练模型进行微调后,得到目标检测模型。本发明极大减少目标检测模型训练时间,同时相较于在单一目标检测数据集训练获得更好的性能。
0/0