基于HG-RTDETR的建筑工地小目标检测方法OA
HG-RTDETR for small object detection in construction sites
针对建筑场景下小目标检测存在的像素信息有限、光照不稳定、目标尺度变化大及遮挡干扰等挑战,研究提出一种基于Transformer架构的改进型检测模型HG-RTDETR.该方法构建了单头门控特征提取模块,通过单头注意力机制实现全局上下文与局部特征的协同建模,并引入门控机制优化特征选择,增强遮挡条件下的特征判别能力.同时设计跨尺度上下文融合网络,其核心包含可重构感受野卷积(RFAConv)与层级特征整合模块(HFI):RFAConv通过自适应调整感受野实现多尺度特征提取并抑制噪声干扰;HFI通过特征金字塔架构融合高层语义、中层细节及大尺度特征,建立跨层级特征交互机制.此外,在边界框回归阶段引入MPDIoU损失函数,通过精细化刻画预测框与真实框的匹配度,提升小目标边界定位的精度与鲁棒性.在MOCS建筑数据集上的实验表明,此方法的mAP50和mAP50:95分别提升2.8%和1.8%.
To address the challenges of limited pixel information,unstable illumination,large-scale variations,and occlusion interference in small-object detection within construction sites,this paper proposes an improved Transformer-based detection model,HG-RTDETR.A Single-Head Gated feature extraction module(SHG)was constructed,employing a single-head attention mechanism to jointly model global context and local features,with a gating mechanism to optimize feature selection and enhance discrimination under occlusion.A Cross-Scale Channel and Context Feature Fusion(CS-CCFF)was designed,incorporating Reconfigurable-Field Attention Convolution(RFAConv)and Hierarchical Feature Integrator(HFI).RFA adaptively adjusts receptive fields for multi-scale fea-ture extraction while suppressing noise,and HFI integrates high-level semantics,mid-level details,and large-scale features via a feature pyramid architecture to establish cross-level feature interaction.In addition,the MPDIoU loss function was introduced during bounding-box regression to refine the alignment between predicted and ground-truth boxes,improving small-object localization accuracy and robustness.Experiments on the MOCS construction dataset demonstrate improvements of 2.8%in mAP50 and 1.8%in mAP50:95.
杨亚龙;崔昊宇;苏亮亮;陈永麟
安徽省智能建筑与建筑节能重点实验室,安徽 合肥 230022||安徽建筑大学 电子信息工程学院,安徽 合肥 230001安徽省智能建筑与建筑节能重点实验室,安徽 合肥 230022||安徽建筑大学 电子信息工程学院,安徽 合肥 230001安徽省智能建筑与建筑节能重点实验室,安徽 合肥 230022||安徽建筑大学 电子信息工程学院,安徽 合肥 230001安徽省智能建筑与建筑节能重点实验室,安徽 合肥 230022||安徽建筑大学 电子信息工程学院,安徽 合肥 230001
建筑与水利
小目标检测RT-DETR跨尺度特征融合门控注意力机制
small object detectionRT-DETRcross-scale feature fusiongated attention mechanism
《苏州科技大学学报(自然科学版)》 2026 (2)
52-61,10
国家重点研发计划项目(2024YFC3808100)安徽建筑大学科研项目(JZ202383JZ202372)
评论