search

dvlab-research / PFENet

PFENet: Prior Guided Feature Enrichment Network for Few-shot Segmentation (TPAMI).
322 stars 53 forks source link

分割结果可视化 #59

Open XBPFS opened 2 years ago

XBPFS commented 2 years ago

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 commented 2 years ago

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