一种基于DETR模型的黄河流域坝岸险情检测方法

摘要

本发明提供一种基于DETR模型的黄河流域坝岸险情检测方法，获取坝岸样本图像集合，坝岸样本图像集合包括黄河流域出险和无险情时的坝岸样本图像；对坝岸样本图像集合进行标注，获取图像训练集和图像测试集；根据图像训练集和图像测试集，对DETR网络模型进行训练，直至训练结束，获取坝岸险情检测模型；获取实时坝岸图像，利用坝岸险情检测模型进行检测，判断是否出现险情。采用DETR模型进行目标检测，DETR模型具有强大的并行计算特性，提升速度，减少训练时间，提高了实时性，该方法有效的移除了传统深度检测方法中的许多手工设计的部件以简化检测流程，可以同时对所有目标进行检测，效率更高。