Single image water hazard detection using FCN with reflection attention units

Xiaofeng Han, Chuong Nguyen, Shaodi You, Jianfeng Lu*

*Corresponding author for this work

    Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

    31 Citations (Scopus)

    Abstract

    Water bodies, such as puddles and flooded areas, on and off road pose significant risks to autonomous cars. Detecting water from moving camera is a challenging task as water surface is highly refractive, and its appearance varies with viewing angle, surrounding scene, weather conditions. In this paper, we present a water puddle detection method based on a Fully Convolutional Network (FCN) with our newly proposed Reflection Attention Units (RAUs). An RAU is a deep network unit designed to embody the physics of reflection on water surface from sky and nearby scene. To verify the performance of our proposed method, we collect 11455 color stereo images with polarizers, and 985 of left images are annotated and divided into 2 datasets: On Road (ONR) dataset and Off Road (OFR) dataset. We show that FCN-8s with RAUs improves significantly precision and recall metrics as compared to FCN-8s, DeepLab V2 and Gaussian Mixture Model (GMM). We also show that focal loss function can improve the performance of FCN-8s network due to the extreme imbalance of water versus ground classification problem.

    Original languageEnglish
    Title of host publicationComputer Vision – ECCV 2018 - 15th European Conference, 2018, Proceedings
    EditorsMartial Hebert, Yair Weiss, Vittorio Ferrari, Cristian Sminchisescu
    PublisherSpringer Verlag
    Pages105-121
    Number of pages17
    ISBN (Print)9783030012304
    DOIs
    Publication statusPublished - 2018
    Event15th European Conference on Computer Vision, ECCV 2018 - Munich, Germany
    Duration: 8 Sept 201814 Sept 2018

    Publication series

    NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
    Volume11210 LNCS
    ISSN (Print)0302-9743
    ISSN (Electronic)1611-3349

    Conference

    Conference15th European Conference on Computer Vision, ECCV 2018
    Country/TerritoryGermany
    CityMunich
    Period8/09/1814/09/18

    Fingerprint

    Dive into the research topics of 'Single image water hazard detection using FCN with reflection attention units'. Together they form a unique fingerprint.

    Cite this