FCL-Net: Towards accurate edge detection via Fine-scale Corrective Learning.

Journal: Neural networks : the official journal of the International Neural Network Society
PMID:

Abstract

Integrating multi-scale predictions has become a mainstream paradigm in edge detection. However, most existing methods mainly focus on effective feature extraction and multi-scale feature fusion while ignoring the low learning capacity in fine-level branches, limiting the overall fusion performance. In light of this, we propose a novel Fine-scale Corrective Learning Net (FCL-Net) that exploits semantic information from deep layers to facilitate fine-scale feature learning. FCL-Net mainly consists of a Top-down Attentional Guiding (TAG) and a Pixel-level Weighting (PW) module. TAG module adopts semantic attentional cues from coarse-scale prediction into guiding the fine-scale branches by learning a top-down LSTM. PW module treats the contribution of each spatial location independently and promote fine-level branches to detect detailed edges with high confidence. Experiments on three benchmark datasets, i.e., BSDS500, Multicue, and BIPED, show that our approach significantly outperforms the baseline and achieves a competitive ODS F-measure of 0.826 on the BSDS500 benchmark. The source code and models are publicly available at https://github.com/DREAMXFAR/FCL-Net.

Authors

  • Wenjie Xuan
    School of Computer Science, Wuhan University, Wuhan, China; National Engineering Research Center for Multimedia Software, Wuhan University, Wuhan, China; Institute of Artificial Intelligence, Wuhan University, Wuhan, China; Hubei Key Laboratory of Multimedia and Network Communication Engineering, Wuhan University, Wuhan, China.
  • Shaoli Huang
    School of Computer Science, Faculty of Engineering, The University of Sydney, Darlington, NSW 2008, Australia.
  • Juhua Liu
    School of Printing and Packaging, Wuhan University, Wuhan, China; Institute of Artificial Intelligence, Wuhan University, Wuhan, China. Electronic address: liujuhua@whu.edu.cn.
  • Bo Du
    School of Computer Science, Wuhan University, Wuhan, 430072, China. Electronic address: remoteking@whu.edu.cn.