Uncertainty quantification for learning-based computational imaging