Skip to content

Deep Learning-based Laser Speckle Reduction

License

Notifications You must be signed in to change notification settings

masontchen/DeepLSR

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

39 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep Laser Speckle Reduction (DeepLSR)

If you use this code, please cite:

Taylor L. Bobrow, Faisal Mahmood, Miguel Inserni, Nicholas J. Durr, “DeepLSR: Deep learning approach for laser speckle reduction" arxiv link

Setup

Prerequisites

  • Linux (Tested on Ubuntu 16.04)
  • NVIDIA GPU (Tested on Nvidia P100 using Google Cloud)
  • CUDA CuDNN (CPU mode and CUDA without CuDNN may work with minimal modification, but untested)
  • Pytorch>=0.4.0
  • torchvision>=0.2.1
  • dominate>=2.3.1
  • visdom>=0.1.8.3
  • scipy

Dataset

The dataset can be downloaded here (7GB).

Dataset Organization

All image pairs must be 256x256 and paired together in 2048x1024 images. '.png' and '.jpg' files are acceptable. Data needs to be arranged in the following order:

SOMEPATH # Some arbitrary path
└── Datasets # Datasets folder
      └── XYZ_Dataset # Active dataset
            ├── test
            └── train

Training

To train a model:

python train.py --dataroot <datapath> --name DeepLSR  --gpu_ids 0 --display_id 0 
--lambda_L1 70 --niter 200 --niter_decay 200 --pool_size 64 --loadSize 1024 --fineSize 1024
  • To view training losses and results, run python -m visdom.server and click the URL http://localhost:8097. For cloud servers replace localhost with your IP.
  • To view epoch-wise intermediate training results, ./checkpoints/DeepLSR/web/index.html
  • --niter number of epochs with constant learning rate --niter_decay number of epochs with linearly decaying learning rate
  • --lr adjusts the learning rate, default = 0.002

Pre-trained Models

Our pre-trained models can be downloaded here.

Testing

To test the model:

python test.py --dataroot <datapath> --name DeepLSR --gpu_ids 0 --display_id 0 
--loadSize 1024 --fineSize 1024
  • The test results will be saved to a html file here: ./results/DeepLSR/test_latest/index.html.

Issues

  • Please open new threads or report issues to faisalm@jhu.edu
  • Immidiate response to minor issues may not be available.

License

© Durr Lab - This project is licensed under the MIT License - see the LICENSE.md file for details

Acknowledgments

  • Subsidized computing resources were provided by Google Cloud.

Reference

If you find our work useful in your research please consider citing our paper:

@inproceedings{bobrow2018deeplsr,
  title     = {DeepLSR: Deep learning approach for laser speckle reduction},
  author    = {Taylor L. Bobrow, Faisal Mahmood, Miguel Inserni, Nicholas J. Durr},
  booktitle = {arXiv.1810.10039},
  year = {2018}
}

About

Deep Learning-based Laser Speckle Reduction

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 97.8%
  • Shell 2.2%