摘要
本发明公开了一种基于FHC‑DETR的废钢分类判级方法,该方法包括:获取待分类的废钢图片并进行预处理;利用训练好的废钢分类判级模型废钢进行分类判级,其中,废钢分类判级模型为基于FHC‑DETR的模型,通过对RT‑DETR模型的改进,生成FHC‑DETR模型,具体包括:在RT‑DETR模型的主干网络中引入FasterNet模块;在高效混合编码器中采用HiLo注意力模块替换原有的AIFI模块,并采用CGR‑FFM模块替换原有的CCFM模块;在FHC‑DETR模型的解码器中,边界框回归损失函数采用基于最小点距离损失的MPDIoU损失函数来替换RT‑DETR中采用的基于GIoU损失函数。本发明的方法提高了在废钢分类判级任务中检测精度和效率。