在审

目标检测方法、装置、设备、存储介质和程序产品

闫胜春、王鑫、韩君龙、马斌、王雷、卞楠、范子文、谢志鹏、史博航
国能朔黄铁路发展有限责任公司

摘要

本申请涉及一种目标检测方法、装置、设备、存储介质和程序产品。该方法包括:获取接触网部件的待检测图像并输入预先训练好的小目标检测模型,通过其中的骨干网络提取与接触网部件相匹配的目标通道的通道注意力特征以及与接触网部件相匹配的空间注意力特征,进而得到特征增强图;并将其输入颈部网络,通过其获取特征增强图中的接触网部件的浅层特征以及深层语义空间特征,并将两个特征进行加权融合,得到多个融合特征图;并将其输入多个头部网络,通过其分别预测各融合特征图中的接触网部件对应的检测框,并根据检测框得到接触网部件的多个检测结果。采用本方法能够提高检测准确率。

暂无引用专利