Line detection via a lightweight CNN with a Hough Layer

Lev Teplyakov,Kirill Kaymakov,Evgeny Shvets,Dmitry Nikolaev
DOI: https://doi.org/10.48550/arXiv.2008.08884
2020-10-16
Abstract:Line detection is an important computer vision task traditionally solved by Hough Transform. With the advance of deep learning, however, trainable approaches to line detection became popular. In this paper we propose a lightweight CNN for line detection with an embedded parameter-free Hough layer, which allows the network neurons to have global strip-like receptive fields. We argue that traditional convolutional networks have two inherent problems when applied to the task of line detection and show how insertion of a Hough layer into the network solves them. Additionally, we point out some major inconsistencies in the current datasets used for line detection.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?