You do not need to use an image map, you can also just use a heightmap Colour by Height output from a Heightfield Generate, or a Convert Displacement to Scalar omitting heightmap generate altogether. That's a step you can leave out altogether and used built in functions.