基于U-net的水位线检测

来源：计量学报杂志2019年第3期北京时间：

作者：程淑红,赵考鹏,张仕军,张典范

单位：燕山大学 电气工程学院, 河北 秦皇岛 066004

摘要：提出一种基于U-net水位线自动分割的新方法,并通过多种场景进行验证。首先标记出原始图像中的水和背景并对其灰度化;然后利用处理后的图像和原始图像制作出数据集,把数据集作为输入利用U-net对图像进行分割;最后将所有分割出来的图像进行边缘提取得到水位线。实验结果表明:U-net水位自动分割可以精确地标记出水位线,同时解决了在水位测量过程中图像背景所带来的影响,分割效果明显优于其它分割方法,识别率达到96%以上。

关键词：计量学;水位检测;数据集;U-net;分割;边界提取

基金资助：国家自然科学基金(61601400);河北省博士后基金(B2016003027);秦皇岛市科学技术研究与发展计划(201701B009)