Please see UQ_DRIVE_stochastic_sample_2000.ipynb
and UQ_DRIVE_stochastic_sample_200.ipynb
.
models.py
contains the scripts for the Bayesian dropout and network architectures (2D U-Net).utils.py
contains functions for the prediction and the preprocessing procedures.
- DRIVE website
- J.J. Staal, M.D. Abramoff, M. Niemeijer, M.A. Viergever, B. van Ginneken, "Ridge based vessel segmentation in color images of the retina", IEEE Transactions on Medical Imaging, 2004, vol. 23, pp. 501-509.