Image Classification
Transformers
Safetensors
vit
vision
STEM-AI-mtl/City_map
Google
ViT
STEM-AI-mtl
Instructions to use STEM-AI-mtl/City_map-vit-base-patch16-224 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use STEM-AI-mtl/City_map-vit-base-patch16-224 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="STEM-AI-mtl/City_map-vit-base-patch16-224") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("STEM-AI-mtl/City_map-vit-base-patch16-224") model = AutoModelForImageClassification.from_pretrained("STEM-AI-mtl/City_map-vit-base-patch16-224", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload ViTForImageClassification
Browse files- README.md +2 -3
- config.json +92 -92
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -1,16 +1,15 @@
|
|
| 1 |
---
|
| 2 |
license: other
|
| 3 |
-
license_name: stem.ai.mtl
|
| 4 |
-
license_link: LICENSE
|
| 5 |
tags:
|
| 6 |
- vision
|
| 7 |
- image-classification
|
| 8 |
- STEM-AI-mtl/City_map
|
| 9 |
- Google
|
| 10 |
- ViT
|
| 11 |
-
|
| 12 |
datasets:
|
| 13 |
- STEM-AI-mtl/City_map
|
|
|
|
|
|
|
| 14 |
widget:
|
| 15 |
- src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
|
| 16 |
example_title: Tiger
|
|
|
|
| 1 |
---
|
| 2 |
license: other
|
|
|
|
|
|
|
| 3 |
tags:
|
| 4 |
- vision
|
| 5 |
- image-classification
|
| 6 |
- STEM-AI-mtl/City_map
|
| 7 |
- Google
|
| 8 |
- ViT
|
|
|
|
| 9 |
datasets:
|
| 10 |
- STEM-AI-mtl/City_map
|
| 11 |
+
license_name: stem.ai.mtl
|
| 12 |
+
license_link: LICENSE
|
| 13 |
widget:
|
| 14 |
- src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
|
| 15 |
example_title: Tiger
|
config.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"_name_or_path": "
|
| 3 |
"architectures": [
|
| 4 |
"ViTForImageClassification"
|
| 5 |
],
|
|
@@ -9,101 +9,101 @@
|
|
| 9 |
"hidden_dropout_prob": 0.0,
|
| 10 |
"hidden_size": 768,
|
| 11 |
"id2label": {
|
| 12 |
-
"0": "
|
| 13 |
-
"1": "
|
| 14 |
-
"2": "
|
| 15 |
-
"3": "
|
| 16 |
-
"4": "
|
| 17 |
-
"5": "
|
| 18 |
-
"6": "
|
| 19 |
-
"7": "
|
| 20 |
-
"8": "
|
| 21 |
-
"9": "
|
| 22 |
-
"10": "
|
| 23 |
-
"11": "
|
| 24 |
-
"12": "
|
| 25 |
-
"13": "
|
| 26 |
-
"14": "
|
| 27 |
-
"15": "
|
| 28 |
-
"16": "
|
| 29 |
-
"17": "
|
| 30 |
-
"18": "
|
| 31 |
-
"19": "
|
| 32 |
-
"20": "
|
| 33 |
-
"21": "
|
| 34 |
-
"22": "
|
| 35 |
-
"23": "
|
| 36 |
-
"24": "
|
| 37 |
-
"25": "
|
| 38 |
-
"26": "
|
| 39 |
-
"27": "
|
| 40 |
-
"28": "
|
| 41 |
-
"29": "
|
| 42 |
-
"30": "
|
| 43 |
-
"31": "
|
| 44 |
-
"32": "
|
| 45 |
-
"33": "
|
| 46 |
-
"34": "
|
| 47 |
-
"35": "
|
| 48 |
-
"36": "
|
| 49 |
-
"37": "
|
| 50 |
-
"38": "
|
| 51 |
-
"39": "
|
| 52 |
-
"40": "
|
| 53 |
-
"41": "
|
| 54 |
-
"42": "
|
| 55 |
-
"43": "
|
| 56 |
-
"44": "
|
| 57 |
},
|
| 58 |
"image_size": 224,
|
| 59 |
"initializer_range": 0.02,
|
| 60 |
"intermediate_size": 3072,
|
| 61 |
"label2id": {
|
| 62 |
-
"Amsterdam":
|
| 63 |
-
"Barcelone":
|
| 64 |
-
"Boston":
|
| 65 |
-
"Budapest":
|
| 66 |
-
"Cairo":
|
| 67 |
-
"Calgary":
|
| 68 |
-
"Charlottetown":
|
| 69 |
-
"Chicago":
|
| 70 |
-
"Dakar":
|
| 71 |
-
"Dubai":
|
| 72 |
-
"Dublin":
|
| 73 |
-
"Edmonton":
|
| 74 |
-
"Fredericton":
|
| 75 |
-
"Halifax":
|
| 76 |
-
"Hong_Kong":
|
| 77 |
-
"Istanbul":
|
| 78 |
-
"Las_Vegas":
|
| 79 |
-
"Londres":
|
| 80 |
-
"Los_Angeles":
|
| 81 |
-
"Melbourne":
|
| 82 |
-
"Miami":
|
| 83 |
-
"Montreal":
|
| 84 |
-
"Mumbai":
|
| 85 |
-
"New_York":
|
| 86 |
-
"Oslo":
|
| 87 |
-
"Ottawa":
|
| 88 |
-
"Paris":
|
| 89 |
-
"Philadelphie":
|
| 90 |
-
"Prague":
|
| 91 |
-
"Quebec":
|
| 92 |
-
"Regina":
|
| 93 |
-
"San_Francisco":
|
| 94 |
-
"Seattle":
|
| 95 |
-
"Seoul":
|
| 96 |
-
"Singapore":
|
| 97 |
-
"St-John":
|
| 98 |
-
"Stockholm":
|
| 99 |
-
"Sydney":
|
| 100 |
-
"Tokyo":
|
| 101 |
-
"Toronto":
|
| 102 |
-
"Vancouver":
|
| 103 |
-
"Venise":
|
| 104 |
-
"Washington":
|
| 105 |
-
"Winnipeg":
|
| 106 |
-
"Zurich":
|
| 107 |
},
|
| 108 |
"layer_norm_eps": 1e-12,
|
| 109 |
"model_type": "vit",
|
|
@@ -114,5 +114,5 @@
|
|
| 114 |
"problem_type": "single_label_classification",
|
| 115 |
"qkv_bias": true,
|
| 116 |
"torch_dtype": "float32",
|
| 117 |
-
"transformers_version": "4.
|
| 118 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"_name_or_path": "google/vit-base-patch16-224",
|
| 3 |
"architectures": [
|
| 4 |
"ViTForImageClassification"
|
| 5 |
],
|
|
|
|
| 9 |
"hidden_dropout_prob": 0.0,
|
| 10 |
"hidden_size": 768,
|
| 11 |
"id2label": {
|
| 12 |
+
"0": "Ottawa",
|
| 13 |
+
"1": "Seoul",
|
| 14 |
+
"2": "Philadelphie",
|
| 15 |
+
"3": "Dakar",
|
| 16 |
+
"4": "Paris",
|
| 17 |
+
"5": "Sydney",
|
| 18 |
+
"6": "Toronto",
|
| 19 |
+
"7": "Montreal",
|
| 20 |
+
"8": "Istanbul",
|
| 21 |
+
"9": "Venise",
|
| 22 |
+
"10": "Regina",
|
| 23 |
+
"11": "Mumbai",
|
| 24 |
+
"12": "Dubai",
|
| 25 |
+
"13": "Seattle",
|
| 26 |
+
"14": "Fredericton",
|
| 27 |
+
"15": "St-John",
|
| 28 |
+
"16": "Edmonton",
|
| 29 |
+
"17": "Los_Angeles",
|
| 30 |
+
"18": "Winnipeg",
|
| 31 |
+
"19": "Chicago",
|
| 32 |
+
"20": "Singapore",
|
| 33 |
+
"21": "Stockholm",
|
| 34 |
+
"22": "Halifax",
|
| 35 |
+
"23": "Barcelone",
|
| 36 |
+
"24": "Londres",
|
| 37 |
+
"25": "Vancouver",
|
| 38 |
+
"26": "Las_Vegas",
|
| 39 |
+
"27": "New_York",
|
| 40 |
+
"28": "Prague",
|
| 41 |
+
"29": "Amsterdam",
|
| 42 |
+
"30": "Calgary",
|
| 43 |
+
"31": "Miami",
|
| 44 |
+
"32": "Washington",
|
| 45 |
+
"33": "Dublin",
|
| 46 |
+
"34": "Charlottetown",
|
| 47 |
+
"35": "Budapest",
|
| 48 |
+
"36": "Boston",
|
| 49 |
+
"37": "Oslo",
|
| 50 |
+
"38": "Zurich",
|
| 51 |
+
"39": "Cairo",
|
| 52 |
+
"40": "Quebec",
|
| 53 |
+
"41": "Tokyo",
|
| 54 |
+
"42": "Melbourne",
|
| 55 |
+
"43": "Hong_Kong",
|
| 56 |
+
"44": "San_Francisco"
|
| 57 |
},
|
| 58 |
"image_size": 224,
|
| 59 |
"initializer_range": 0.02,
|
| 60 |
"intermediate_size": 3072,
|
| 61 |
"label2id": {
|
| 62 |
+
"Amsterdam": 29,
|
| 63 |
+
"Barcelone": 23,
|
| 64 |
+
"Boston": 36,
|
| 65 |
+
"Budapest": 35,
|
| 66 |
+
"Cairo": 39,
|
| 67 |
+
"Calgary": 30,
|
| 68 |
+
"Charlottetown": 34,
|
| 69 |
+
"Chicago": 19,
|
| 70 |
+
"Dakar": 3,
|
| 71 |
+
"Dubai": 12,
|
| 72 |
+
"Dublin": 33,
|
| 73 |
+
"Edmonton": 16,
|
| 74 |
+
"Fredericton": 14,
|
| 75 |
+
"Halifax": 22,
|
| 76 |
+
"Hong_Kong": 43,
|
| 77 |
+
"Istanbul": 8,
|
| 78 |
+
"Las_Vegas": 26,
|
| 79 |
+
"Londres": 24,
|
| 80 |
+
"Los_Angeles": 17,
|
| 81 |
+
"Melbourne": 42,
|
| 82 |
+
"Miami": 31,
|
| 83 |
+
"Montreal": 7,
|
| 84 |
+
"Mumbai": 11,
|
| 85 |
+
"New_York": 27,
|
| 86 |
+
"Oslo": 37,
|
| 87 |
+
"Ottawa": 0,
|
| 88 |
+
"Paris": 4,
|
| 89 |
+
"Philadelphie": 2,
|
| 90 |
+
"Prague": 28,
|
| 91 |
+
"Quebec": 40,
|
| 92 |
+
"Regina": 10,
|
| 93 |
+
"San_Francisco": 44,
|
| 94 |
+
"Seattle": 13,
|
| 95 |
+
"Seoul": 1,
|
| 96 |
+
"Singapore": 20,
|
| 97 |
+
"St-John": 15,
|
| 98 |
+
"Stockholm": 21,
|
| 99 |
+
"Sydney": 5,
|
| 100 |
+
"Tokyo": 41,
|
| 101 |
+
"Toronto": 6,
|
| 102 |
+
"Vancouver": 25,
|
| 103 |
+
"Venise": 9,
|
| 104 |
+
"Washington": 32,
|
| 105 |
+
"Winnipeg": 18,
|
| 106 |
+
"Zurich": 38
|
| 107 |
},
|
| 108 |
"layer_norm_eps": 1e-12,
|
| 109 |
"model_type": "vit",
|
|
|
|
| 114 |
"problem_type": "single_label_classification",
|
| 115 |
"qkv_bias": true,
|
| 116 |
"torch_dtype": "float32",
|
| 117 |
+
"transformers_version": "4.39.2"
|
| 118 |
}
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 343356252
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bf3d0663d7c175c8da59e82d22ba1982f8e3c74e395b89106b910775ff83ae3a
|
| 3 |
size 343356252
|