请问如何可视化推理所得的掩码图像？

BBBBchan / CorrMatch

Official code for "CorrMatch: Label Propagation via Correlation Matching for Semi-Supervised Semantic Segmentation"

117 stars 8 forks source link

Closed hindinbush closed 3 months ago

hindinbush commented 3 months ago

您好，非常感谢您出色的工作，请问在训练结束之后，如何获得预测所得的掩码图像？

BBBBchan commented 3 months ago

您好，感谢您的关注。我们已经上传了掩码图像可视化的代码。您可以在vis_mask.py中修改checkpoint和保存路径信息，并运行sh tools/vis_mask.sh <gpu num> <port>进行可视化。

hindinbush commented 3 months ago

您好，感谢您的关注。我们已经上传了掩码图像可视化的代码。您可以在vis_mask.py中修改checkpoint和保存路径信息，并运行sh tools/vis_mask.sh <gpu num> <port>进行可视化。

非常感谢您的回复，这对我帮助很大，后续有什么不懂得地方我可能还得再麻烦您解答一下哈哈，感谢！