You can try adding a custom AOV, and then create a aiUtility node, change shade mode to flat color, and color mode to normal. Plug that into your custom AOV's shader option. Picked this up from this tutorial: https://www.artstation.com/artwork/ZLGNG (at the very bottom) Good luck!