(self, dataset=None)
return labelcats, ['object', 'part', 'material']
Lists label and category names.
Lists label and category names.
Lists label and category names .
def get_label_and_category_names(self, dataset=None):
        '''
        Lists label and category names.
        '''
        # Labels are ordered as follows:
        # 0, [object labels] [divided object labels] [materials] [parts]
        # The zero label is reserved to mean 'no prediction'.
        suffixes = []
        divided_labels = []
        for suffix in suffixes:
            divided_labels.extend([('%s-%s' % (label, suffix), 'part')
                for label in self.segmodel.labeldata['object'][1:]])
        # Create the whole list of labels
        labelcats = (
                [(label, 'object')
                    for label in self.segmodel.labeldata['object']] +
                divided_labels +
                [(label, 'material')
                    for label in self.segmodel.labeldata['material'][1:]] +
                [(label, 'part') for label in self.part_names])
        return labelcats, ['object', 'part', 'material']
def get_label_and_category_names ( self , dataset = None ) : # Labels are ordered as follows: # 0, [object labels] [divided object labels] [materials] [parts] # The zero label is reserved to mean 'no prediction'. suffixes = [ ] divided_labels = [ ] for suffix in suffixes : divided_labels . extend ( [ ( '%s-%s' % ( label , suffix ) , 'part' ) for label in self . segmodel . labeldata [ 'object' ] [ 1 : ] ] ) # Create the whole list of labels labelcats = ( [ ( label , 'object' ) for label in self . segmodel . labeldata [ 'object' ] ] + divided_labels + [ ( label , 'material' ) for label in self . segmodel . labeldata [ 'material' ] [ 1 : ] ] + [ ( label , 'part' ) for label in self . part_names ] ) return labelcats , [ 'object' , 'part' , 'material' ]