commit files to HF hub

.gitattributes

@@ -40,3 +40,4 @@ images/l_shaped_kitchen.jpeg filter=lfs diff=lfs merge=lfs -text
 images/others.jpeg filter=lfs diff=lfs merge=lfs -text
 images/single_wall_kitchen.jpeg filter=lfs diff=lfs merge=lfs -text
 images/u_shaped_kitchen.jpeg filter=lfs diff=lfs merge=lfs -text
+images/l_and_u_shaped_kitchen.jpeg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -21,4 +21,20 @@ model-index:
 # kitchen-layouts-2.2.0
 
 
-Kitchen Layouts detection🤗🖼️
+Kitchen Layouts detection🤗🖼️
+
+
+## Example Images
+
+
+#### island_kitchen
+
+![island_kitchen](images/island_kitchen.jpeg)
+
+#### l_and_u_shaped_kitchen
+
+![l_and_u_shaped_kitchen](images/l_and_u_shaped_kitchen.jpeg)
+
+#### single_wall_kitchen
+
+![single_wall_kitchen](images/single_wall_kitchen.jpeg)

config.json

@@ -9,23 +9,17 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "g_shaped_kitchen",
-    "1": "galley_kitchen",
-    "2": "island_kitchen",
-    "3": "l_shaped_kitchen",
-    "4": "single_wall_kitchen",
-    "5": "u_shaped_kitchen"
+    "0": "island_kitchen",
+    "1": "l_and_u_shaped_kitchen",
+    "2": "single_wall_kitchen"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "g_shaped_kitchen": 0,
-    "galley_kitchen": 1,
-    "island_kitchen": 2,
-    "l_shaped_kitchen": 3,
-    "single_wall_kitchen": 4,
-    "u_shaped_kitchen": 5
+    "island_kitchen": "0",
+    "l_and_u_shaped_kitchen": "1",
+    "single_wall_kitchen": "2"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -36,5 +30,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.39.0"
+  "transformers_version": "4.39.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b99a3e854df44e58bdd71865e8f9e87775dd2d992404f4162c2ffecdd165b030
-size 343236280
+oid sha256:c14ba3eba7ae49cc30d11f996315a8fcb1e3b0af0a30570cb778e7ea805b8c48
+size 343227052

preprocessor_config.json

@@ -21,7 +21,7 @@
     0.5,
     0.5
   ],
-  "image_processor_type": "ViTImageProcessor",
+  "image_processor_type": "ViTFeatureExtractor",
   "image_std": [
     0.5,
     0.5,