How to use the code 'IOUEval.py' in 'train' directory?

sacmehta / ESPNet

ESPNet: Efficient Spatial Pyramid of Dilated Convolutions for Semantic Segmentation

https://sacmehta.github.io/ESPNet/

MIT License

541 stars 112 forks source link

How to use the code 'IOUEval.py' in 'train' directory? #25

Closed wangq95 closed 5 years ago

wangq95 commented 5 years ago

The code 'IOUEval.py' defined the class of mIoU and per class iou calculation, and used in training procedure, but how to reuse this code for evaluating the predicted result? Thank you!

sacmehta commented 5 years ago

IOUEval file just need predicted segmentation mask and ground truth. If you have this information you can easily call the function inside IOUEval to compute mIOU