Official implementation for "Compressive video via IR-pulsed illumination"
- Python 3.9
- Pytorch >=1.10+
- Numpy
- Scikit-image
- Scikit-learn
- tqdm
- scipy
- mpmath
- tensorboard
- install anaconda (https://www.anaconda.com/products/distribution)
- on anaconda prompt (windows) or terminal (linux) create enviroment:
conda create -n dpwfs python=3.9
conda activate dpwfs
- Install pytorch + cuda:
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
This codes is used to train the proposed model with the DAVIS2017 dataset
If you find our project useful, please cite:
@article{guzman2023compressive,
title={Compressive video via IR-pulsed illumination},
author={Guzm{\'a}n, Felipe and Skowronek, James and Vera, Esteban and Brady, David J},
journal={Optics Express},
volume={31},
number={23},
pages={39201--39212},
year={2023},
publisher={Optica Publishing Group}
}
The codes are based on STFormer, we also refer to codes in FILM, EDSC. Thanks for sharing the code!.
You can contact Felipe Guzman by sending mail to [email protected] for questions regarding this repository