Gray prediction image before encode

[shadydiaa]

i want to save the Gray prediction image before encode it to color , how can i modify the code to save gray image and why not if show the classes prediction with final colored image ? .. really thank you

[hellochick]

You can get the output from raw_output_up instead of pred.