chunked/nltk_chunking/tasks_semantic_segmentation/chunk_62.txt

image.size[::-1],
 )
pred_seg = tf.math.argmax(upsampled_logits, axis=-1)[0]

To visualize the results, load the dataset color palette as ade_palette() that maps each class to their RGB values.