Home

Awesome

<p align=center> A Lightweight Road Detection Algorithm Based on Multiscale Convolutional Attention Network and Coupled Decoder Head (IEEE GRSL 2023) </p>

<p align=center> Aerial Remote Sensing Image Cascaded Road Detection Network Based on Edge Sensing Module and Attention Module (IEEE GRSL 2022) </p>

This repository contains python implementation of our paper LRDNet and CasEANet.

1. Required environments:

2. Use LRDNet：

Download the dataset and Pretrained Model of LRDNet

Baidu Pan：https://pan.baidu.com/s/1HT0RzsF-zMVhYuq0Bce0tg?pwd=5ous code：5ous

config file:

The config file in /configs/LRDNet_RNBD.yml is the model of LRDNet.

The config file in configs\CaeEANet_RNBD.yml is the model of CasEANet.

train the model：train.py

python train.py --config ... --work-dir ... --log_interval ... --save_every ... --resume_from ... --device ...

analysisi the complexity：analysisi_model.py

python analysisi_model.py --config ... --size ...

predict a patch image ：demo.py

python demo.py --config ... --checkpoints ... --path_img ...

predict a large image：predict.py

python predict.py --config ... --checkpoints ... --path_img ... --path_save ... --device

browser_dataset：browser_dataset.py

python browser_dataset.py --config ... --wait_time ...

3. Visual Result

<p align="center"> <img src="predict/RNBD/8.png"/> <br /> </p> <p align="center"> <img src="predict/result/predict_surf/8.png"/> <br /> </p>

4. Citation

Please cite our paper if you find the work useful:

@ARTICLE{10098603,
  author={Liu, Dongyang and Zhang, Junping and Qi, Yunxiao and Zhang, Ye},
  journal={IEEE Geoscience and Remote Sensing Letters}, 
  title={A Lightweight Road Detection Algorithm Based on Multiscale Convolutional Attention Network and Coupled Decoder Head}, 
  year={2023},
  volume={20},
  number={},
  pages={1-5},
  doi={10.1109/LGRS.2023.3266054}}

@ARTICLE{9829847,
  author={Liu, Dongyang and Zhang, Junping and Liu, Kun and Zhang, Ye},
  journal={IEEE Geoscience and Remote Sensing Letters}, 
  title={Aerial Remote Sensing Image Cascaded Road Detection Network Based on Edge Sensing Module and Attention Module}, 
  year={2022},
  volume={19},
  number={},
  pages={1-5},
  doi={10.1109/LGRS.2022.3190495}}