(self,opt=None,idx2color_path='../../backup/idx2color.txt',num_class=32)
| 5 | |
| 6 | class SegHelper: |
| 7 | def __init__(self,opt=None,idx2color_path='../../backup/idx2color.txt',num_class=32): |
| 8 | self.opt = opt |
| 9 | self.num_classes = num_class |
| 10 | self.idx2color_path = idx2color_path |
| 11 | f = open(self.idx2color_path, 'r') |
| 12 | self.idx2color = {k:[] for k in range(self.num_classes)} |
| 13 | for j in range(256): |
| 14 | line = f.readline() |
| 15 | line = line.strip(' \n').strip('[').strip(']').strip(' ').split() |
| 16 | line = [int(l) for l in line if l.isdigit()] |
| 17 | self.idx2color[j] = line # color in rgb order |
| 18 | |
| 19 | self.color2idx = {tuple(v):k for k,v in self.idx2color.items()} |
| 20 | name2idx = Dict_indexing() |
| 21 | self.name2idx = {k: name2idx[k]['index'] for k in name2idx.keys()} |
| 22 | self.idx2name = {v:k for k,v in self.name2idx.items()} |
| 23 | self.idx2name_padding = {v:'BG' for v in range(self.num_classes,256)} |
| 24 | self.idx2name.update(self.idx2name_padding) |
| 25 | |
| 26 | def unique(self,array): |
| 27 | uniq, index = np.unique(array, return_index=True, axis=0) |
nothing calls this directly
no test coverage detected