| 109 | |
| 110 | |
| 111 | def colorEncode(labelmap, colors, mode='RGB'): |
| 112 | labelmap = labelmap.astype('int') |
| 113 | labelmap_rgb = np.zeros((labelmap.shape[0], labelmap.shape[1], 3), |
| 114 | dtype=np.uint8) |
| 115 | for label in unique(labelmap): |
| 116 | if label < 0: |
| 117 | continue |
| 118 | labelmap_rgb += (labelmap == label)[:, :, np.newaxis] * \ |
| 119 | np.tile(colors[label], |
| 120 | (labelmap.shape[0], labelmap.shape[1], 1)) |
| 121 | |
| 122 | if mode == 'BGR': |
| 123 | return labelmap_rgb[:, :, ::-1] |
| 124 | else: |
| 125 | return labelmap_rgb |
| 126 | |
| 127 | |
| 128 | def accuracy(preds, label): |