Visualization of classification results

[gqy122]

Hello author, after carefully reading your paper, I successfully ran through the code for your paper. However, the visualization of the classification image results is different from some other papers. How can I convert this classification result graph in your paper into a regular classification result graph

[qichaoliu]

Hello. Once you get the predicted labels for each pixel, you can set a specific color for each label in the classification map to present it in any way you want.