(self, segmentations, label=None)
return segviz.seg_as_image(segmentations, size=self.size)
Visualizes the given segmentation tensor.  If label is specified,
        zeros all labels other than the given one.
Visualizes the given segmentation tensor.  If label is specified,
        zeros all labels other than the given one.
Visualizes the given segmentation tensor . If label is specified zeros all labels other than the given one .
def segmentation(self, segmentations, label=None):
        '''
        Visualizes the given segmentation tensor.  If label is specified,
        zeros all labels other than the given one.
        '''
        if label is not None:
            segmentations = segmentations.clone()
            segmentations[segmentations != label] = 0
        return segviz.seg_as_image(segmentations, size=self.size)
def segmentation ( self , segmentations , label = None ) : if label is not None : segmentations = segmentations . clone ( ) segmentations [ segmentations != label ] = 0 return segviz . seg_as_image ( segmentations , size = self . size )