diff --git a/README.md b/README.md index 09e8966..10c6a9f 100644 --- a/README.md +++ b/README.md @@ -1,8 +1,10 @@ # L-CNN — End-to-End Wireframe Parsing +This repository contains the official PyTorch implementation of the paper: *Yichao Zhou, Haozhi Qi, Yi Ma. ["End-to-End Wireframe Parsing."](https://arxiv.org/abs/1905.03246) arXiv:1905.03246 [cs.CV]*. + ## Introduction -This repository contains the official PyTorch implementation of [L-CNN](https://arxiv.org/abs/1905.03246), a conceptually simple yet effective neural network-based algorithm for detecting the wireframe from a given image. It outperforms the previous state-of-the-art wireframe and line extraction algorithms by a large margin. We hope that this repository serves as an easily reproducible baseline for future researches in this area. +[L-CNN](https://arxiv.org/abs/1905.03246) is a conceptually simple yet effective neural network for detecting the wireframe from a given image. It outperforms the previous state-of-the-art wireframe and line detectors by a large margin. We hope that this repository serves as an easily reproducible baseline for future researches in this area. ## Main Results @@ -14,7 +16,7 @@ This repository contains the official PyTorch implementation of [L-CNN](https:// ### Quantitative Measures -The following table reports the performance of several wireframe and line detection algorithms on the [Wireframe dataset](https://github.com/huangkuns/wireframe). +The following table reports the performance metrics of several wireframe and line detectors on the [Wireframe dataset](https://github.com/huangkuns/wireframe). | | Wireframe (sAP10) | Wireframe (APH) | Wireframe (FH) | Wireframe (mAPJ) | | :--------------------------------------------------: | :--------------------------------: | :-----------------------------: | :----------------------------: | :------------------------------: | @@ -63,6 +65,7 @@ eval-APH.py # script for APH evaluation eval-mAPJ.py # script for mAPJ evaluation train.py # script for training the neural network post.py # script for post-processing +process.py # script for processing a dataset from a checkpoint ``` ## Reproducing Results @@ -93,7 +96,7 @@ rm *.xz cd .. ``` -If `gdrive-download.sh` does not work for you, you can download the data manually from [Google +If `gdrive-download.sh` does not work for you, you can download the pre-processed dataset manually from [Google Drive](https://drive.google.com/drive/u/1/folders/1rXLAh5VIj8jwf8vLfuZncStihRO2chFr) and proceed accordingly.