| 49 | return seg_map_class |
| 50 | |
| 51 | def classmap2colormap(self,seg_map_class): |
| 52 | seg_map_class_flattened = seg_map_class.view(-1,1) |
| 53 | seg_map_color_flattened = torch.zeros(seg_map_class.shape[0]*seg_map_class.shape[1],3).cuda().long() |
| 54 | for cls, color in self.idx2color.items(): |
| 55 | matching_indices = (seg_map_class_flattened == torch.LongTensor([cls]).cuda()) |
| 56 | seg_map_color_flattened[matching_indices.view(-1)] = torch.LongTensor(color).cuda() |
| 57 | seg_map_color_flattened = seg_map_color_flattened.view(seg_map_class.shape[0],seg_map_class.shape[1],3) |
| 58 | return seg_map_color_flattened |
| 59 | |
| 60 | def split_SemAndChange(self,seg_map_class): |
| 61 | seg_map_change_class = seg_map_class//50 |