tfwu / hawp

Holistically-Attracted Wireframe Parsing (CVPR 2020)

This is the offical implementation for our CVPR paper.

Highlights

• We propose a fast and parsimonious parsing method HAWP to accurately and robustly detect a vectorized wireframe in an input image with a single forward pass.
• The proposed HAWP is fully end-to-end.
• The proposed HAWP does not require squeeze module.
• State-of-the-art performance on the Wireframe dataset and YorkUrban dataset.
• The proposed HAWP achievs 29.5 FPS on a GPU (Tesla V100) for 1-batch inference.

Quantitative Results

Wireframe Dataset

Method	Wireframe Dataset							FPS
	sAP5	sAP10	sAP15	msAP	mAPJ	APH	FH
LSD	/	/	/	/	/	55.2	62.5	49.6
AFM	18.5	24.4	27.5	23.5	23.3	69.2	77.2	13.5
DWP	3.7	5.1	5.9	4.9	40.9	67.8	72.2	2.24
L-CNN	58.9	62.9	64.9	62.2	59.3	80.3	76.9	15.6
						82.8†	81.3†
L-CNN (re-trained)	59.7	63.6	65.3	62.9	60.2	81.6	77.9	15.6
						83.7†	81.7†
HAWP (Ours)	62.5	66.5	68.2	65.7	60.2	84.5	80.3	29.5
						86.1†	83.1†

YorkUrban Dataset

Method	YorkUrban Dataset							FPS
	sAP5	sAP10	sAP15	msAP	mAPJ	APH	FH
LSD	/	/	/	/	/	50.9	60.1	49.6
AFM	7.3	9.4	11.1	9.3	12.4	48.2	63.3	13.5
DWP	1.5	2.1	2.6	2.1	13.4	51.0	61.6	2.24
L-CNN	24.3	26.4	27.5	26.1	30.4	58.5	61.8	15.6
						59.6†	65.3†
L-CNN (re-trained)	25.0	27.1	28.3	26.8	31.5	58.3	62.2	15.6
						59.3†	65.2†
HAWP (Ours)	26.1	28.5	29.7	28.1	31.6	60.6	64.8	29.5
						61.2†	66.3†

Installation & Pretrained Model

Will be finished in this week.

Citations

If you find our work useful in your research, please consider citing:

@inproceedings{HAWP,
title = "Holistically-Attracted Wireframe Parsing",
author = "Nan Xue and Tianfu Wu and Song Bai and Fu-Dong Wang and Gui-Song Xia and Liangpei Zhang and Philip H.S. Torr
",
booktitle = "IEEE Conference on Computer Vision and Pattern Recognition (CVPR)",
year = {2020},
}

Acknoledgement

We acknowledge the effort from the authors of the Wireframe dataset and the YorkUrban dataset. These datasets make accurate line segment detection and wireframe parsing possible.