Firstly, I think the `** 2` is specific to 2D. But secondly, I don't think this is necessary — I think the weighting is fully captured by multiplying the color channel by the ratio... See discussion in #666.