There is no way to convert LA into color as you cannot infer color from grayscale. I don’t know which task are you performing but if you want to use RGB pictures are grountruth you need 3-channek outputs. If you get wrong outputs you may have to rethink the pipeline