一种基于污水排放图像数据自动标注的污水检测方法

本发明公开了一种基于污水排放图像数据自动标注的污水检测方法，包括如下步骤：获取污水排放的数据集，并对获取的数据集进行预处理以及将获取的数据集按照7:3的比例划分为训练集和测试集；构建对比学习网络模型，将训练集送入至对比学习网络模型中训练，得到训练好的对比学习网络模型；将训练好的对比学习网络模型与YOLO5检测网络进行拼接获得混合网络模型，并对混合网络模型进行训练，以获得训练好的混合网络模型；将测试集送入混合网络模型中，得到最终的检测结果，本发明目的在于替代人工，以实现图像数据的自动标注以及自动检测，用以实现自动化获取污水检测结果。