inis 23(1): e4

Research Article

A Fully Convolutional Network with Waterfall Atrous Spatial Pooling and Localized Active Contour Loss for Fish Segmentation

Download274 downloads
  • @ARTICLE{10.4108/eetinis.v10i1.2942,
        author={Thanh Viet Le and Van Yem Vu and Van Truong Pham and Thi-Thao Tran},
        title={A Fully Convolutional Network with Waterfall Atrous Spatial Pooling and Localized Active Contour Loss for Fish Segmentation},
        journal={EAI Endorsed Transactions on Industrial Networks and Intelligent Systems},
        volume={10},
        number={1},
        publisher={EAI},
        journal_a={INIS},
        year={2023},
        month={4},
        keywords={Fully convolutional network, DeepFish, SIUM fish data, Waterfall Atrous Spatial Pooling},
        doi={10.4108/eetinis.v10i1.2942}
    }
    
  • Thanh Viet Le
    Van Yem Vu
    Van Truong Pham
    Thi-Thao Tran
    Year: 2023
    A Fully Convolutional Network with Waterfall Atrous Spatial Pooling and Localized Active Contour Loss for Fish Segmentation
    INIS
    EAI
    DOI: 10.4108/eetinis.v10i1.2942
Thanh Viet Le1, Van Yem Vu1, Van Truong Pham1, Thi-Thao Tran1,*
  • 1: Hanoi University of Science and Technology
*Contact email: thao.tranthi@hust.edu.vn

Abstract

Accurate measurements and statistics of fish data are important for sustainable development of aqua-enviroment and marine fisheries. For data measurements and statistics, automatic segmentation of fish is one of key tasks. The fish segmentation however is a challenging task due to arterfacts in underwater images. In this study, we introduce a deep-learning approach, namely FCN-WRN-WASP for automatic fish segmentation from the underwater images. In particular, we introduce a computational-efficient variation called Waterfall Atrous Spatial Pooling (WASP) module into a Fully convolutional network with Wide ResNet baseline. We also proposed a loss function inspired from active contour approach that can exploit the local intensity information from the input image. The approach has been validated on the DeepFish data and the SIUM data set. The results are promissing for fish segmentation, with higher Intersection over Union (IoU) scores compared to state of the arts. The evaluation results showed that the incorporation of the image based active contour loss helps increase the segmentation performance. In addition, the use of the WASP in the architecture is effective especially for forground fish segmentation.