Convert the heatmap to the result image

[yiyi-today]

Hey, I see that you have output the resulting graph, but the result is heatmap in the test.py，and I want to know how did you convert it to the result image。Hope to get your answer, thank you very much！

[bolin-sys]

Hey, I see that you have output the resulting graph, but the result is heatmap in the test.py，and I want to know how did you convert it to the result image。Hope to get your answer, thank you very much！

I meet this question too, and are you deal this question? Thanks very much!