RANK++LETR: Learn to Rank and Optimize Candidates for Line Segment Detection
========


## Pretrained Model
Pretrained model can be download [here](https://drive.google.com/file/d/14aY80zrVuLg9aIJNYW3ipuQgsP51VRUt/view?usp=sharing).


## Data preparation
Download the Wireframe dataset [here](https://github.com/huangkuns/wireframe) and resize the images and the ground truth to 512x512.


## Evaluation
To evaluate RANK++LETR on the Wireframe dataset with a single GPU run:
```
python train_line.py --batch_size 2 --num_workers=2 --test --dataset_file=wireframe --resume /path/to/xx.pth
```
The same performance as the paper should be obtained

## Visualization Results
More visualization results of the RANK++LETR can be found in the supp_show.pdf