commit files to HF hub

.gitattributes

@@ -48,3 +48,8 @@ images/hallway_room.jpg filter=lfs diff=lfs merge=lfs -text
 images/home_office_room.jpg filter=lfs diff=lfs merge=lfs -text
 images/kitchen_room.jpg filter=lfs diff=lfs merge=lfs -text
 images/laundry_room_staging.jpg filter=lfs diff=lfs merge=lfs -text
+images/balcony.jpg filter=lfs diff=lfs merge=lfs -text
+images/floorplan.jpg filter=lfs diff=lfs merge=lfs -text
+images/neighbourhood.jpg filter=lfs diff=lfs merge=lfs -text
+images/office_room.jpg filter=lfs diff=lfs merge=lfs -text
+images/yard.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -15,7 +15,7 @@ model-index:
     metrics:
       - name: Accuracy
         type: accuracy
-        value: 0.7663043737411499
+        value: 0.7490494251251221
 ---
 
 # house_rooms
@@ -31,6 +31,10 @@ Report any issues with the demo at the [github repo](https://github.com/nateraw/
 ## Example Images
 
 
+#### balcony
+
+![balcony](images/balcony.jpg)
+
 #### bathroom staging
 
 ![bathroom staging](images/bathroom_staging.jpg)
@@ -39,21 +43,17 @@ Report any issues with the demo at the [github repo](https://github.com/nateraw/
 
 ![bedroom staging](images/bedroom_staging.jpg)
 
-#### building facade
-
-![building facade](images/building_facade.jpg)
-
 #### dining room
 
 ![dining room](images/dining_room.jpg)
 
-#### hallway room
+#### floorplan
 
-![hallway room](images/hallway_room.jpg)
+![floorplan](images/floorplan.jpg)
 
-#### home office room
+#### hallway room
 
-![home office room](images/home_office_room.jpg)
+![hallway room](images/hallway_room.jpg)
 
 #### kitchen room
 
@@ -65,4 +65,16 @@ Report any issues with the demo at the [github repo](https://github.com/nateraw/
 
 #### living room
 
-![living room](images/living_room.jpg)
+![living room](images/living_room.jpg)
+
+#### neighbourhood
+
+![neighbourhood](images/neighbourhood.jpg)
+
+#### office room
+
+![office room](images/office_room.jpg)
+
+#### yard
+
+![yard](images/yard.jpg)

config.json

@@ -9,29 +9,35 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "bathroom staging",
-    "1": "bedroom staging",
-    "2": "building facade",
+    "0": "balcony",
+    "1": "bathroom staging",
+    "10": "office room",
+    "11": "yard",
+    "2": "bedroom staging",
     "3": "dining room",
-    "4": "hallway room",
-    "5": "home office room",
+    "4": "floorplan",
+    "5": "hallway room",
     "6": "kitchen room",
     "7": "laundry room staging",
-    "8": "living room"
+    "8": "living room",
+    "9": "neighbourhood"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "bathroom staging": "0",
-    "bedroom staging": "1",
-    "building facade": "2",
+    "balcony": "0",
+    "bathroom staging": "1",
+    "bedroom staging": "2",
     "dining room": "3",
-    "hallway room": "4",
-    "home office room": "5",
+    "floorplan": "4",
+    "hallway room": "5",
     "kitchen room": "6",
     "laundry room staging": "7",
-    "living room": "8"
+    "living room": "8",
+    "neighbourhood": "9",
+    "office room": "10",
+    "yard": "11"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:726d993a0890579f986c4da12a629125decd6d1dd5c7417ebac416ec02fd18be
-size 343287469
+oid sha256:8a7be941ad8d86dd632ab85cf3aa822b5042d54306bc967ef3381234f87af949
+size 343296685

runs/events.out.tfevents.1692044701.bbec58bd2423.186.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7512944f1b6fd9616074dbfeb04247257ca311685e6d316b5e649a7ecac6e28f
+size 5200