基于U-net的水位線檢測

摘要：提出一種基于U-net水位線自動分割的新方法,并通過多種場景進行驗證。首先標記出原始圖像中的水和背景并對其灰度化;然后利用處理后的圖像和原始圖像制作出數據集,把數據集作為輸入利用U-net對圖像進行分割;最后將所有分割出來的圖像進行邊緣提取得到水位線。實驗結果表明:U-net水位自動分割可以精確地標記出水位線,同時解決了在水位測量過程中圖像背景所帶來的影響,分割效果明顯優于其它分割方法,識別率達到96%以上。