Structure-aware computational imaging