Official implementation of "Single Image Depth Estimation Trained via Depth from Defocus Cues" (arxiv). The implementation is based on the architectures of DeepLabV3+ and Self-Attention.
In order to build the Gaussian PSF layer on your own machine, run the following line
python ext/setup.py install
Download the official dataset and use SfmLearner for data preparation.
Download the official dataset and unpack into Test and Train folders.
Download the official dataset and use the following preprocessing instructions.
Download the official dataset from here and generate all bokeh images accorfing to Srinivasan.
Download the official dataset from here.
CUDA_VISIBLE_DEVICES=0 python train.py --nof-focus 2