传输外线资源识别检测方法、装置、设备及存储介质
摘要:
本发明公开了传输外线资源识别检测方法、装置、设备及存储介质,其中,该方法包括:将传输外线资源图片标注为训练数据集和测试数据集,并利用训练数据集训练得到目标检测模型;将测试数据集输入至目标检测模型中,以利用骨干网络对测试数据集进行图像特征提取得到第一图像特征;将第一图像特征和位置编码矩阵输入至编码器进行编码以输出编码向量,并将对象查询集和编码向量输入至解码器模块进行解码操作得到解码输出结果;将解码输出结果输入至检测头进行测试数据集的图片中的目标类别检测和边界框预测,以输出模型的识别检测结果。本发明将本场景的目标检测问题转化为集合预测问题使得模型在不依赖非极大值抑制的情况下进行有效的端到端训练。
0/0