How to output colorized range images in Semantic-Map-to-Lidar task?

[HITysx]

Thank you for your excellent work! Could you tell me how to output colorized range image samples in Semantic-Map-to-Lidar task? In other words, how to add semantic labels to output range images in Semantic-Map-to-Lidar task?