wkentaro / pytorch-fcn

PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)
MIT License
1.74k stars 480 forks source link

fcn.utils.visualize_segmentation,where the visualize_segmentation function? #164

Closed liul-ing closed 2 years ago

liul-ing commented 2 years ago

"viz = fcn.utils.visualize_segmentation(lbl_pred=lp, lbl_true=lt, img=img, n_class=n_class) " in the validate function in trainer.py,i can not find the visulaize_segmentation function.could you please tell me about my questions?

wkentaro commented 2 years ago

Please follow the instruction. https://github.com/wkentaro/pytorch-fcn#installation