yyliu01 / PS-MT

[CVPR'22] Perturbed and Strict Mean Teachers for Semi-supervised Semantic Segmentation
https://arxiv.org/pdf/2111.12903.pdf
MIT License
186 stars 17 forks source link

visualization #34

Closed 836469383 closed 1 year ago

836469383 commented 1 year ago

First of all, the author is really good. This is a great job. Then the rookie wants a visual script. Any help will be greatly appreciated!

yyliu01 commented 1 year ago

Hi @836469383,

The visualisation can be performed via matplotlib package in python.

import matplotlib.pyplot as plt
# for the label y with dimension H x W
plt.imshow(y)
plt.show()

# for the prediction with dimension class x H x W
plt.imshow(torch.argmax(y_hat, dim=0))
plt.show()

Cheers, Yuyuan

836469383 commented 1 year ago

Thanks a lot! And sorry to bother you, but there's one small Q: what is the VT, and where can I find its code? thanks again image

yyliu01 commented 1 year ago

Hi @836469383 The AT denotes the assistance teaching, which means the ensemble prediction for the pseudo-label, refers to the code in here.

836469383 commented 1 year ago

谢谢带佬,祝您工作顺利,生活愉快,祝好~

yyliu01 commented 1 year ago

老哥不用客气