Is there any way to bake Material ID map to use with software like Quixel Suite? Like the one I attached below. It could be baked by taking highpoly mesh material exported to .obj material library, and then write its color to the baked texture. Maybe there's another, existing way to bake this map, I don't know.
