Update config.json

config.json

@@ -1,20 +1,227 @@
 {
   "name": "nicpras_finetuned_yolo",
   "framework": "torch",
-  "labels": [
-    "Person", "Scooter", "Wheelchair", "Phone", "Cup", "Book", "Cloth",
-    "Shoes", "Bicycle", "Skating Shoes", "Toy", "Headphone", "Food",
-    "Bottle", "Ball", "Map", "Ice cream", "Camera", "Shovel",
-    "Sandcastle bucket", "Chair", "Trolley", "Swimming float", "Slide",
-    "Water", "Plate", "Tray", "Wallet", "Ipad", "Luggage", "Car",
-    "Package", "Microphone", "Cleaning trolley", "Stroller", "Clothes",
-    "Flyers", "Water can", "Compost", "Pot", "Plant", "Pitcher", "Bin",
-    "Bucket", "Aquarium", "Fishing net", "Sandcastle", "Newspaper", "Shell",
-    "Sunscreen", "Water gun", "Bookshelf", "Yarn", "Baby"
-  ],
   "type": "segmentation",
-  "pipeline_tag": "object-detection",
+  "pipeline_tag": "image-segmentation",
   "params": {
     "threshold": 0.5
-  }
-}
+  },
+  "labels": [
+    {
+      "name": "Person",
+      "attributes": []
+    },
+    {
+      "name": "Scooter",
+      "attributes": []
+    },
+    {
+      "name": "Wheelchair",
+      "attributes": []
+    },
+    {
+      "name": "Phone",
+      "attributes": []
+    },
+    {
+      "name": "Cup",
+      "attributes": []
+    },
+    {
+      "name": "Book",
+      "attributes": []
+    },
+    {
+      "name": "Cloth",
+      "attributes": []
+    },
+    {
+      "name": "Shoes",
+      "attributes": []
+    },
+    {
+      "name": "Bicycle",
+      "attributes": []
+    },
+    {
+      "name": "Skating Shoes",
+      "attributes": []
+    },
+    {
+      "name": "Toy",
+      "attributes": []
+    },
+    {
+      "name": "Headphone",
+      "attributes": []
+    },
+    {
+      "name": "Food",
+      "attributes": []
+    },
+    {
+      "name": "Bottle",
+      "attributes": []
+    },
+    {
+      "name": "Ball",
+      "attributes": []
+    },
+    {
+      "name": "Map",
+      "attributes": []
+    },
+    {
+      "name": "Ice cream",
+      "attributes": []
+    },
+    {
+      "name": "Camera",
+      "attributes": []
+    },
+    {
+      "name": "Shovel",
+      "attributes": []
+    },
+    {
+      "name": "Sandcastle bucket",
+      "attributes": []
+    },
+    {
+      "name": "Chair",
+      "attributes": []
+    },
+    {
+      "name": "Trolley",
+      "attributes": []
+    },
+    {
+      "name": "Swimming float",
+      "attributes": []
+    },
+    {
+      "name": "Slide",
+      "attributes": []
+    },
+    {
+      "name": "Water",
+      "attributes": []
+    },
+    {
+      "name": "Plate",
+      "attributes": []
+    },
+    {
+      "name": "Tray",
+      "attributes": []
+    },
+    {
+      "name": "Wallet",
+      "attributes": []
+    },
+    {
+      "name": "Ipad",
+      "attributes": []
+    },
+    {
+      "name": "Luggage",
+      "attributes": []
+    },
+    {
+      "name": "Car",
+      "attributes": []
+    },
+    {
+      "name": "Package",
+      "attributes": []
+    },
+    {
+      "name": "Microphone",
+      "attributes": []
+    },
+    {
+      "name": "Cleaning trolley",
+      "attributes": []
+    },
+    {
+      "name": "Stroller",
+      "attributes": []
+    },
+    {
+      "name": "Clothes",
+      "attributes": []
+    },
+    {
+      "name": "Flyers",
+      "attributes": []
+    },
+    {
+      "name": "Water can",
+      "attributes": []
+    },
+    {
+      "name": "Compost",
+      "attributes": []
+    },
+    {
+      "name": "Pot",
+      "attributes": []
+    },
+    {
+      "name": "Plant",
+      "attributes": []
+    },
+    {
+      "name": "Pitcher",
+      "attributes": []
+    },
+    {
+      "name": "Bin",
+      "attributes": []
+    },
+    {
+      "name": "Bucket",
+      "attributes": []
+    },
+    {
+      "name": "Aquarium",
+      "attributes": []
+    },
+    {
+      "name": "Fishing net",
+      "attributes": []
+    },
+    {
+      "name": "Sandcastle",
+      "attributes": []
+    },
+    {
+      "name": "Newspaper",
+      "attributes": []
+    },
+    {
+      "name": "Shell",
+      "attributes": []
+    },
+    {
+      "name": "Sunscreen",
+      "attributes": []
+    },
+    {
+      "name": "Water gun",
+      "attributes": []
+    },
+    {
+      "name": "Bookshelf",
+      "attributes": []
+    },
+    {
+      "name": "Yarn",
+      "attributes": []
+    },
+    {
+      "name": "Baby",
+      "attributes": []
+    }
+  ]
+}