Awesome

<img height="23" src="https://raw.githubusercontent.com/lh9171338/Outline/master/icon.jpg"/> FE-LSD

This repository contains the official PyTorch implementation of the paper: Detecting Line Segments in Motion-blurred Images with Events.

Introduction

FE-LSD is an event-enhanced line segment detection framework for motion-blurred images with thoughtful information fusion of both modalities and advanced wireframe parsing network. Extensive results on both synthetic and realistic datasets demonstrate the effectiveness of the proposed method for handling motion blurs in line segment detection.

Network Architecture

Results

FE-Wireframe Dataset

• Quantitative Comparisons

• Qualitative Comparisons

FE-Blurframe Dataset

• Quantitative Comparisons

• Qualitative Comparisons

Requirements

• torch>=1.6.0
• torchvision>=0.7.0
• CUDA>=10.1
• lh_tool, matplotlib, numpy, opencv_python, Pillow, scikit_learn, scipy, setuptools, tensorboardX, timm, torch, torchvision, tqdm, yacs,

Step-by-step installation

conda create --name FE-LSD python=3.8
conda activate FE-LSD

cd <FE-LSD-Path>
git clone https://github.com/lh9171338/FE-LSD.git
cd FE-LSD

pip install -r requirements.txt

python setup.py build_ext --inplace

Quickstart with the pretrained model

• There are pretrained models in Google drive and Baiduyun. Please download them and put in the model/ folder.
• Put your test data in the dataset/ folder and generate the test.json file.

python image2json.py --dataset_name <DATASET_NAME>

• The file structure is as follows:

|-- dataset
    |-- events
	|-- 000001.npz
	|-- ...
    |-- images-blur
	|-- 000001.png
	|-- ...
    |-- test.json

• Test with the pretrained model. The results are saved in the output/ folder.

python test.py --arch <ARCH> --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> --save_image

Training & Testing

Data Preparation

• Download the dataset from Baiduyun.
• Unzip the dataset to the dataset/ folder.
• Convert event streams into synchronous frames using Event Spike Tensor (EST) representation.

python event2frame.py --dataset_name <DATASET_NAME> --representation EST

# for FE-Wireframe dataset
python event2frame.py --dataset_name FE-Wireframe --representation EST

# for FE-Blurframe dataset
python event2frame.py --dataset_name FE-Blurframe --representation EST -s 0.5

ln -s events-EST-10 events

Train

python train.py --arch FE-HAWP --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> [--gpu <GPU_ID>] # FE-HAWP
python train.py --arch FE-ULSD --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> [--gpu <GPU_ID>] # FE-ULSD

Test

python test.py --arch FE-HAWP --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> --save_image --with_clear [--gpu <GPU_ID>] # FE-HAWP
python test.py --arch FE-ULSD --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> --save_image --with_clear [--gpu <GPU_ID>] # FE-ULSD

Evaluation

To evaluate the mAP^J, sAP, and FPS

python test.py --arch FE-HAWP --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> --evaluate [--gpu <GPU_ID>] # FE-HAWP
python test.py --arch FE-ULSD --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> --evaluate [--gpu <GPU_ID>] # FE-ULSD

To evaluate AP^H and F^H, MATLAB is required

cd metric
python eval_APH.py --arch FE-HAWP --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> # FE-HAWP
python eval_APH.py --arch FE-ULSD --dataset_name <DATASET_NAME> --model_name <MODEL_NAME> # FE-ULSD

Citation

@ARTICLE{10323537,
  author={Yu, Huai and Li, Hao and Yang, Wen and Yu, Lei and Xia, Gui-Song},
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence}, 
  title={Detecting Line Segments in Motion-Blurred Images With Events}, 
  year={2023},
  pages={1-16},
  doi={10.1109/TPAMI.2023.3334877}
}