Add function to return a model of the diffuse structure in the image.