摘要
本发明提供一种轨道小目标检测方法、装置、电子设备及存储介质,涉及人工智能技术领域,方法包括:获取待检测图像;将待检测图像输入至预训练的小目标检测模型,获得小目标检测模型输出的小目标检测结果;小目标检测模型是基于样本检测图像和样本检测图像对应的样本小目标检测结果,对基于改进的yolo11算法构建的初始模型进行预训练得到的;小目标检测模型,用于对待检测图像进行特征提取,生成图像特征,对图像特征进行特征对齐和特征融合,生成融合特征,并基于融合特征进行小目标检测。通过上述方式,可准确检测出待检测图像中的行人或落石等小目标,提高轨道小目标检测结果的准确性。