分割结果可视化

[XBPFS]

Hi,Thank you for your work .
I tried to follow this method https://github.com/hszhao/semseg/blob/master/tool/test.py#L216 to get the segmentation result , but due to the random horizontal rotation and mirroring processing set in the experiment, the obtained results cannot be superimposed on the original image.And I can't find any other solution because I'm too clumsy。
Could you please share the visualization code you use, it would be greatly appreciated。

[tianzhuotao]

When doing the visualization, you can implement you customized transforms to avoid the resize and crop operations.