摘要
本发明涉及冶金技术领域,公开一种基于逐层方向指导的个性化联邦学习废钢检测方法,包括如下步骤:废钢图片采集及预处理:通过高分辨率工业相机在各废钢回收站的不同卸料点采集不同种类废钢的高清图片作为训练数据集;对采集到的图像数据进行标注。采用基于逐层方向指导机制的个性化联邦学习,通过客户端模型间的差异,利用逐层方向指导机制计算各客户端在层级上的方向向量,用以指导各客户端个性化模型的更新方向,以此为各废钢回收站点定制个性化模型,使个性化模型可在保证兼顾学习其他部分废钢回收站点数据的情况下,进一步针对自身本地数据进行精细的学习,提升模型在自身客户端上的性能。