(self,img_seg)
| 33 | return unique_colors |
| 34 | |
| 35 | def extract_class_from_seg(self,img_seg): |
| 36 | unique_colors = self.extract_color_from_seg(img_seg) # (num_class_in_img,3) # color channel in rgb order |
| 37 | classes_idx = [self.color2idx[tuple(color.tolist())]for color in unique_colors] |
| 38 | classes_str = [self.idx2name[idx] for idx in classes_idx] |
| 39 | return classes_idx, classes_str |
| 40 | |
| 41 | def colormap2classmap(self,seg_array): |
| 42 | seg_array_flattened = torch.LongTensor(seg_array.reshape(-1,3)).cuda() |
no test coverage detected