Pytorch Image Segmentation Metrics