A method for depth mapping includes capturing an electronic image of a scene using an imaging device. The electronic image is processed to generate depth data with respect to the scene. The gain of the imaging device is set responsively to the depth data.