Can this code be used for gray images?

[YiLiM1]

I have a depth map. Can I use this code?

[JordanMakesMaps]

You should be able to, if you have issues with the number of channels you could duplicate them along the 3rd-axis.