(gray, palette)
| 191 | |
| 192 | |
| 193 | def colorize(gray, palette): |
| 194 | # gray: numpy array of the label and 1*3N size list palette |
| 195 | color = Image.fromarray(gray.astype(np.uint8)).convert('P') |
| 196 | color.putpalette(palette) |
| 197 | return color |
| 198 | |
| 199 | |
| 200 | def find_free_port(): |