modify

app.py

@@ -5,172 +5,174 @@ import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image
 import tensorflow as tf
-from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
-import requests
+from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
+import requests
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
     "nvidia/segformer-b5-finetuned-ade-640-640"
 )
-model = SegformerForSemanticSegmentation.from_pretrained(
+model = TFSegformerForSemanticSegmentation.from_pretrained(
     "nvidia/segformer-b5-finetuned-ade-640-640"
 )
 
+
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [
-        [204, 87, 92],
-        [112, 185, 212],
-        [45, 189, 106],
-        [234, 123, 67],
-        [78, 56, 123],
-        [210, 32, 89],
-        [90, 180, 56],
-        [155, 102, 200],
-        [33, 147, 176],
-        [255, 183, 76],
-        [67, 123, 89],
-        [190, 60, 45],
-        [134, 112, 200],
-        [56, 45, 189],
-        [200, 56, 123],
-        [87, 92, 204],
-        [120, 56, 123],
-        [45, 78, 123],
-        [156, 200, 56],
-        [32, 90, 210],
-        [56, 123, 67],
-        [180, 56, 123],
-        [123, 67, 45],
-        [45, 134, 200],
-        [67, 56, 123],
-        [78, 123, 67],
-        [32, 210, 90],
-        [45, 56, 189],
-        [123, 56, 123],
-        [56, 156, 200],
-        [189, 56, 45],
-        [112, 200, 56],
-        [56, 123, 45],
-        [200, 32, 90],
-        [123, 45, 78],
-        [200, 156, 56],
-        [45, 67, 123],
-        [56, 45, 78],
-        [45, 56, 123],
-        [123, 67, 56],
-        [56, 78, 123],
-        [210, 90, 32],
-        [123, 56, 189],
-        [45, 200, 134],
-        [67, 123, 56],
-        [123, 45, 67],
-        [90, 32, 210],
-        [200, 45, 78],
-        [32, 210, 90],
-        [45, 123, 67],
-        [165, 42, 87],
-        [72, 145, 167],
-        [15, 158, 75],
-        [209, 89, 40],
-        [32, 21, 121],
-        [184, 20, 100],
-        [56, 135, 15],
-        [128, 92, 176],
-        [1, 119, 140],
-        [220, 151, 43],
-        [41, 97, 72],
-        [148, 38, 27],
-        [107, 86, 176],
-        [21, 26, 136],
-        [174, 27, 90],
-        [91, 96, 204],
-        [108, 50, 107],
-        [27, 45, 136],
-        [168, 200, 52],
-        [7, 102, 27],
-        [42, 93, 56],
-        [140, 52, 112],
-        [92, 107, 168],
-        [17, 118, 176],
-        [59, 50, 174],
-        [206, 40, 143],
-        [44, 19, 142],
-        [23, 168, 75],
-        [54, 57, 189],
-        [144, 21, 15],
-        [15, 176, 35],
-        [107, 19, 79],
-        [204, 52, 114],
-        [48, 173, 83],
-        [11, 120, 53],
-        [206, 104, 28],
-        [20, 31, 153],
-        [27, 21, 93],
-        [11, 206, 138],
-        [112, 30, 83],
-        [68, 91, 152],
-        [153, 13, 43],
-        [25, 114, 54],
-        [92, 27, 150],
-        [108, 42, 59],
-        [194, 77, 5],
-        [145, 48, 83],
-        [7, 113, 19],
-        [25, 92, 113],
-        [60, 168, 79],
-        [78, 33, 120],
-        [89, 176, 205],
-        [27, 200, 94],
-        [210, 67, 23],
-        [123, 89, 189],
-        [225, 56, 112],
-        [75, 156, 45],
-        [172, 104, 200],
-        [15, 170, 197],
-        [240, 133, 65],
-        [89, 156, 112],
-        [214, 88, 57],
-        [156, 134, 200],
-        [78, 57, 189],
-        [200, 78, 123],
-        [106, 120, 210],
-        [145, 56, 112],
-        [89, 120, 189],
-        [185, 206, 56],
-        [47, 99, 28],
-        [112, 189, 78],
-        [200, 112, 89],
-        [89, 145, 112],
-        [78, 106, 189],
-        [112, 78, 189],
-        [156, 112, 78],
-        [28, 210, 99],
-        [78, 89, 189],
-        [189, 78, 57],
-        [112, 200, 78],
-        [189, 47, 78],
-        [205, 112, 57],
-        [78, 145, 57],
-        [200, 78, 112],
-        [99, 89, 145],
-        [200, 156, 78],
-        [57, 78, 145],
-        [78, 57, 99],
-        [57, 78, 145],
-        [145, 112, 78],
-        [78, 89, 145],
-        [210, 99, 28],
-        [145, 78, 189],
-        [57, 200, 136],
-        [89, 156, 78],
-        [145, 78, 99],
-        [99, 28, 210],
-        [189, 78, 47],
-        [28, 210, 99],
-        [141, 21, 43],
+        [215, 252, 54],
+        [219, 99, 20],
+        [30, 125, 246],
+        [21, 211, 22],
+        [117, 165, 201],
+        [122, 2, 6],
+        [52, 144, 140],
+        [136, 36, 114],
+        [208, 249, 44],
+        [210, 245, 157],
+        [48, 222, 84],
+        [175, 182, 112],
+        [117, 9, 240],
+        [153, 38, 30],
+        [75, 225, 231],
+        [232, 170, 70],
+        [154, 35, 115],
+        [45, 61, 35],
+        [73, 144, 2],
+        [54, 80, 136],
+        [143, 200, 212],
+        [75, 104, 98],
+        [17, 211, 27],
+        [205, 195, 241],
+        [234, 251, 104],
+        [33, 174, 95],
+        [160, 174, 99],
+        [141, 26, 157],
+        [84, 247, 88],
+        [19, 248, 198],
+        [4, 217, 155],
+        [204, 163, 16],
+        [148, 209, 143],
+        [211, 97, 65],
+        [19, 4, 131],
+        [40, 196, 45],
+        [39, 64, 20],
+        [166, 107, 50],
+        [108, 103, 78],
+        [188, 11, 213],
+        [24, 156, 152],
+        [230, 162, 223],
+        [30, 126, 220],
+        [74, 10, 238],
+        [186, 128, 227],
+        [83, 188, 220],
+        [9, 132, 231],
+        [96, 99, 79],
+        [196, 139, 187],
+        [117, 122, 171],
+        [0, 156, 220],
+        [243, 249, 189],
+        [243, 245, 211],
+        [103, 146, 83],
+        [237, 144, 197],
+        [35, 151, 20],
+        [15, 61, 139],
+        [78, 223, 132],
+        [120, 49, 9],
+        [67, 160, 234],
+        [183, 244, 210],
+        [245, 161, 139],
+        [57, 70, 189],
+        [105, 150, 31],
+        [219, 85, 49],
+        [206, 81, 97],
+        [30, 171, 92],
+        [251, 42, 67],
+        [121, 183, 220],
+        [221, 33, 43],
+        [8, 96, 100],
+        [76, 149, 53],
+        [29, 201, 129],
+        [7, 213, 227],
+        [143, 93, 153],
+        [205, 35, 110],
+        [37, 94, 142],
+        [131, 157, 110],
+        [215, 166, 147],
+        [164, 94, 252],
+        [179, 108, 233],
+        [35, 157, 209],
+        [145, 252, 241],
+        [155, 60, 40],
+        [70, 25, 44],
+        [53, 83, 133],
+        [150, 42, 191],
+        [142, 245, 58],
+        [150, 198, 69],
+        [0, 139, 86],
+        [123, 212, 143],
+        [210, 166, 191],
+        [148, 194, 130],
+        [35, 213, 154],
+        [203, 139, 93],
+        [59, 86, 45],
+        [9, 50, 169],
+        [207, 118, 246],
+        [200, 82, 65],
+        [37, 75, 120],
+        [237, 99, 63],
+        [168, 145, 190],
+        [225, 48, 16],
+        [17, 184, 115],
+        [224, 124, 15],
+        [148, 167, 47],
+        [162, 25, 116],
+        [154, 90, 36],
+        [185, 247, 43],
+        [183, 138, 202],
+        [64, 96, 117],
+        [187, 140, 140],
+        [121, 116, 188],
+        [252, 251, 162],
+        [85, 50, 40],
+        [209, 241, 228],
+        [30, 41, 95],
+        [246, 217, 64],
+        [151, 149, 197],
+        [117, 42, 205],
+        [26, 248, 30],
+        [28, 224, 232],
+        [228, 89, 96],
+        [198, 44, 113],
+        [220, 68, 218],
+        [59, 85, 210],
+        [24, 230, 191],
+        [145, 192, 181],
+        [132, 189, 92],
+        [47, 29, 128],
+        [11, 245, 204],
+        [182, 79, 207],
+        [42, 64, 187],
+        [72, 4, 37],
+        [105, 67, 133],
+        [86, 27, 200],
+        [243, 211, 40],
+        [150, 136, 40],
+        [3, 192, 172],
+        [34, 96, 149],
+        [32, 108, 56],
+        [128, 10, 137],
+        [94, 211, 108],
+        [78, 250, 243],
+        [6, 74, 205],
+        [6, 7, 38],
+        [161, 26, 40],
+        [145, 254, 27],
+        [119, 145, 127],
+        [13, 82, 153],
     ]
 
+
 labels_list = []
 
 with open(r'labels.txt', 'r') as fp:
@@ -179,6 +181,7 @@ with open(r'labels.txt', 'r') as fp:
 
 colormap = np.asarray(ade_palette())
 
+
 def label_to_color_image(label):
     if label.ndim != 2:
         raise ValueError("Expect 2-D input label")
@@ -187,6 +190,7 @@ def label_to_color_image(label):
         raise ValueError("label value too large.")
     return colormap[label]
 
+
 def draw_plot(pred_img, seg):
     fig = plt.figure(figsize=(20, 15))
 
@@ -208,6 +212,7 @@ def draw_plot(pred_img, seg):
     ax.tick_params(width=0.0, labelsize=25)
     return fig
 
+
 def sepia(input_img):
     input_img = Image.fromarray(input_img)
 
@@ -234,11 +239,11 @@ def sepia(input_img):
     fig = draw_plot(pred_img, seg)
     return fig
 
+
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
                     examples=["image-1.jpg", "image-2.jpg", "image-3.jpg", "image-4.jpeg", "image-5.jpg"],
                     allow_flagging='never')
 
-
 demo.launch()