ContourGAN: Image Contour Detection with Generative Adversarial Network

Hongju Yang,Yao Li,Xuefeng Yan,Fuyuan Cao
DOI: https://doi.org/10.1016/j.knosys.2018.09.033
IF: 8.139
2019-01-01
Knowledge-Based Systems
Abstract:We propose a convolutional encoder–decoder framework to extract image contours supported by a generative adversarial network to improve the contour quality. Traditional image-to-image models only consider the loss between prediction and ground truth, neglecting the similarity between the data distribution of the outcomes and ground truth. Based on this observation, the proposed generative adversarial network aims to increase the detection accuracy. The resulting method contains two models, namely an encoder–decoder model, whose weights are updated using binary cross-entropy loss with fine-tuning from the VGG16 pre-trained model, and a discriminator network that employs ground truths and predicted contours as input for discrimination. We evaluate the method on the Berkeley Segmentation DataSetand Benchmarks 500 and NYUD dataset, achieving state-of-the-art performance with an ODS F-measure of 0.810 and 0.715.
What problem does this paper attempt to address?