This repository contains the official PyTorch implementation of the paper: “Learning to Detect Line Segment from Events”

• Installation
• Testing Pre-trained Models
• Downloading the Dataset
• Distillation

For the ease of reproducibility, you are suggested to install miniconda (or anaconda if you prefer) before following executing the following commands.

git clone https://github.com/Qiuben/EvLSD-IED

cd EvLSD-IED

You can download the pretrained model on E-wirferame from xxx.

You can download the synthetic dataset E-wireframe as well as the real-scene datset RE-LSD from OneDrive. You can directly download the EST representation or download the raw event data and convert it to frame representations using the event2frame.py

python train.py -m image -d E-wireframe

python train_IED.py -m event -d E-wireframe