Upload ViTForImageClassification

README.md

@@ -1,16 +1,15 @@
 ---
 license: other
-license_name: stem.ai.mtl
-license_link: LICENSE
 tags:
 - vision
 - image-classification
 - STEM-AI-mtl/City_map
 - Google
 - ViT
-
 datasets:
 - STEM-AI-mtl/City_map
+license_name: stem.ai.mtl
+license_link: LICENSE
 widget:
 - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
   example_title: Tiger

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "STEM-AI-mtl/City_map-vit-base-patch16-224",
+  "_name_or_path": "google/vit-base-patch16-224",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -9,101 +9,101 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "Seattle",
-    "1": "San_Francisco",
-    "2": "Venise",
-    "3": "Mumbai",
-    "4": "St-John",
-    "5": "Istanbul",
-    "6": "Charlottetown",
-    "7": "Miami",
-    "8": "Las_Vegas",
-    "9": "Toronto",
-    "10": "Cairo",
-    "11": "Edmonton",
-    "12": "Los_Angeles",
-    "13": "Barcelone",
-    "14": "Budapest",
-    "15": "Londres",
-    "16": "Singapore",
-    "17": "Chicago",
-    "18": "Quebec",
-    "19": "Stockholm",
-    "20": "Melbourne",
-    "21": "Zurich",
-    "22": "Calgary",
-    "23": "Dakar",
-    "24": "Boston",
-    "25": "Prague",
-    "26": "Dubai",
-    "27": "Philadelphie",
-    "28": "Vancouver",
-    "29": "New_York",
-    "30": "Winnipeg",
-    "31": "Fredericton",
-    "32": "Regina",
-    "33": "Tokyo",
-    "34": "Hong_Kong",
-    "35": "Washington",
-    "36": "Ottawa",
-    "37": "Sydney",
-    "38": "Oslo",
-    "39": "Dublin",
-    "40": "Paris",
-    "41": "Amsterdam",
-    "42": "Montreal",
-    "43": "Halifax",
-    "44": "Seoul"
+    "0": "Ottawa",
+    "1": "Seoul",
+    "2": "Philadelphie",
+    "3": "Dakar",
+    "4": "Paris",
+    "5": "Sydney",
+    "6": "Toronto",
+    "7": "Montreal",
+    "8": "Istanbul",
+    "9": "Venise",
+    "10": "Regina",
+    "11": "Mumbai",
+    "12": "Dubai",
+    "13": "Seattle",
+    "14": "Fredericton",
+    "15": "St-John",
+    "16": "Edmonton",
+    "17": "Los_Angeles",
+    "18": "Winnipeg",
+    "19": "Chicago",
+    "20": "Singapore",
+    "21": "Stockholm",
+    "22": "Halifax",
+    "23": "Barcelone",
+    "24": "Londres",
+    "25": "Vancouver",
+    "26": "Las_Vegas",
+    "27": "New_York",
+    "28": "Prague",
+    "29": "Amsterdam",
+    "30": "Calgary",
+    "31": "Miami",
+    "32": "Washington",
+    "33": "Dublin",
+    "34": "Charlottetown",
+    "35": "Budapest",
+    "36": "Boston",
+    "37": "Oslo",
+    "38": "Zurich",
+    "39": "Cairo",
+    "40": "Quebec",
+    "41": "Tokyo",
+    "42": "Melbourne",
+    "43": "Hong_Kong",
+    "44": "San_Francisco"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Amsterdam": 41,
-    "Barcelone": 13,
-    "Boston": 24,
-    "Budapest": 14,
-    "Cairo": 10,
-    "Calgary": 22,
-    "Charlottetown": 6,
-    "Chicago": 17,
-    "Dakar": 23,
-    "Dubai": 26,
-    "Dublin": 39,
-    "Edmonton": 11,
-    "Fredericton": 31,
-    "Halifax": 43,
-    "Hong_Kong": 34,
-    "Istanbul": 5,
-    "Las_Vegas": 8,
-    "Londres": 15,
-    "Los_Angeles": 12,
-    "Melbourne": 20,
-    "Miami": 7,
-    "Montreal": 42,
-    "Mumbai": 3,
-    "New_York": 29,
-    "Oslo": 38,
-    "Ottawa": 36,
-    "Paris": 40,
-    "Philadelphie": 27,
-    "Prague": 25,
-    "Quebec": 18,
-    "Regina": 32,
-    "San_Francisco": 1,
-    "Seattle": 0,
-    "Seoul": 44,
-    "Singapore": 16,
-    "St-John": 4,
-    "Stockholm": 19,
-    "Sydney": 37,
-    "Tokyo": 33,
-    "Toronto": 9,
-    "Vancouver": 28,
-    "Venise": 2,
-    "Washington": 35,
-    "Winnipeg": 30,
-    "Zurich": 21
+    "Amsterdam": 29,
+    "Barcelone": 23,
+    "Boston": 36,
+    "Budapest": 35,
+    "Cairo": 39,
+    "Calgary": 30,
+    "Charlottetown": 34,
+    "Chicago": 19,
+    "Dakar": 3,
+    "Dubai": 12,
+    "Dublin": 33,
+    "Edmonton": 16,
+    "Fredericton": 14,
+    "Halifax": 22,
+    "Hong_Kong": 43,
+    "Istanbul": 8,
+    "Las_Vegas": 26,
+    "Londres": 24,
+    "Los_Angeles": 17,
+    "Melbourne": 42,
+    "Miami": 31,
+    "Montreal": 7,
+    "Mumbai": 11,
+    "New_York": 27,
+    "Oslo": 37,
+    "Ottawa": 0,
+    "Paris": 4,
+    "Philadelphie": 2,
+    "Prague": 28,
+    "Quebec": 40,
+    "Regina": 10,
+    "San_Francisco": 44,
+    "Seattle": 13,
+    "Seoul": 1,
+    "Singapore": 20,
+    "St-John": 15,
+    "Stockholm": 21,
+    "Sydney": 5,
+    "Tokyo": 41,
+    "Toronto": 6,
+    "Vancouver": 25,
+    "Venise": 9,
+    "Washington": 32,
+    "Winnipeg": 18,
+    "Zurich": 38
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -114,5 +114,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.36.2"
+  "transformers_version": "4.39.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b1a5547134ce458c6a05092bc439c3447d1cee652bb4489d375d6d9138cbb804
+oid sha256:bf3d0663d7c175c8da59e82d22ba1982f8e3c74e395b89106b910775ff83ae3a
 size 343356252