upload v1667704347 model

.gitattributes

@@ -31,4 +31,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-test-image.jpg filter=lfs diff=lfs merge=lfs -text
+test-image1.png filter=lfs diff=lfs merge=lfs -text
+test-image2.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -4,18 +4,16 @@ import tensorflow as tf
 from huggingface_hub import from_pretrained_keras
 from PIL import Image
 
-MODEL_CKPT = "chansung/segmentation-training-pipeline@v1667662600"
+MODEL_CKPT = "chansung/segmentation-training-pipeline@v1667704347"
 MODEL = from_pretrained_keras(MODEL_CKPT)
 
 RESOLTUION = 128
 
-RESOLTUION = 128
-
-ADE_PALETTE = []
+PETS_PALETTE = []
 with open(r"./palette.txt", "r") as fp:
     for line in fp:
         tmp_list = list(map(int, line[:-1].strip("][").split(", ")))
-        ADE_PALETTE.append(tmp_list)
+        PETS_PALETTE.append(tmp_list)
 
 
 def preprocess_input(image: Image) -> tf.Tensor:
@@ -36,7 +34,7 @@ def get_seg_overlay(image, seg):
     color_seg = np.zeros(
         (seg.shape[0], seg.shape[1], 3), dtype=np.uint8
     )  # height, width, 3
-    palette = np.array(ADE_PALETTE)
+    palette = np.array(PETS_PALETTE)
 
     for label, color in enumerate(palette):
         color_seg[seg == label, :] = color
@@ -88,7 +86,7 @@ demo = gr.Interface(
     allow_flagging="never",
     title=title,
     description=description,
-    examples=[["test-image.jpg"]],
+    examples=[["test-image1.png"], ["test-image2.png"]],
 )
 
 demo.launch()

palette.txt

@@ -1,35 +1,3 @@
-[0, 0, 0]
-[216, 82, 24]
-[255, 255, 0]
-[125, 46, 141]
-[118, 171, 47]
-[161, 19, 46]
-[255, 0, 0]
-[0, 128, 128]
-[190, 190, 0]
-[0, 255, 0]
-[0, 0, 255]
-[170, 0, 255]
-[84, 84, 0]
-[84, 170, 0]
-[84, 255, 0]
-[170, 84, 0]
-[170, 170, 0]
-[170, 255, 0]
-[255, 84, 0]
-[255, 170, 0]
-[255, 255, 0]
-[33, 138, 200]
-[0, 170, 127]
-[0, 255, 127]
-[84, 0, 127]
-[84, 84, 127]
-[84, 170, 127]
-[84, 255, 127]
-[170, 0, 127]
-[170, 84, 127]
-[170, 170, 127]
-[170, 255, 127]
-[255, 0, 127]
-[255, 84, 127]
-[255, 170, 127]
+[0, 10, 146]
+[38, 0, 44]
+[255, 232, 0]