一种基于计算机视觉的河道污染识别与预警方法

本发明公开一种基于计算机视觉的河道污染识别与预警方法，包括以下步骤：步骤1，制作河道水质图像数据集；步骤2，构建基于注意力机制的改进ResNet‑18模型；步骤3，构建训练集、验证集和测试集；步骤4，利用训练集训练构建好的改进ResNet‑18模型，再利用验证集调整模型的超参数后，将测试集输入到训练好的深度学习网络中进行性能评估；步骤5，保存模型并进行模型部署，使用训练好的改进ResNet‑18模型进行河道水质污染检测，当将读取的河道图像判别为污染水质时，触发报警信号；本发明实现河道污染的实时监测与早期告警；具有非接触、可监测整个河道、成本低等优点，能第一时间发现偷排行为。