bsvaz's picture
Add Vision Transformer for landmark classification
5c183a0 verified
{
"_name_or_path": "landmark-classification-vit",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Haleakala National Park",
"1": "Mount Rainier National Park",
"2": "Ljubljana Castle",
"3": "Dead Sea",
"4": "Wroclaws Dwarves",
"5": "London Olympic Stadium",
"6": "Niagara Falls",
"7": "Stonehenge",
"8": "Grand Canyon",
"9": "Golden Gate Bridge",
"10": "Edinburgh Castle",
"11": "Mount Rushmore National Memorial",
"12": "Kantanagar Temple",
"13": "Yellowstone National Park",
"14": "Terminal Tower",
"15": "Central Park",
"16": "Eiffel Tower",
"17": "Changdeokgung",
"18": "Delicate Arch",
"19": "Vienna City Hall",
"20": "Matterhorn",
"21": "Taj Mahal",
"22": "Moscow Raceway",
"23": "Externsteine",
"24": "Soreq Cave",
"25": "Banff National Park",
"26": "Pont du Gard",
"27": "Seattle Japanese Garden",
"28": "Sydney Harbour Bridge",
"29": "Petronas Towers",
"30": "Brooklyn Bridge",
"31": "Washington Monument",
"32": "Hanging Temple",
"33": "Sydney Opera House",
"34": "Great Barrier Reef",
"35": "Monumento a la Revolucion",
"36": "Badlands National Park",
"37": "Atomium",
"38": "Forth Bridge",
"39": "Gateway of India",
"40": "Stockholm City Hall",
"41": "Machu Picchu",
"42": "Death Valley National Park",
"43": "Gullfoss Falls",
"44": "Trevi Fountain",
"45": "Temple of Heaven",
"46": "Great Wall of China",
"47": "Prague Astronomical Clock",
"48": "Whitby Abbey",
"49": "Temple of Olympian Zeus"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Atomium": 37,
"Badlands National Park": 36,
"Banff National Park": 25,
"Brooklyn Bridge": 30,
"Central Park": 15,
"Changdeokgung": 17,
"Dead Sea": 3,
"Death Valley National Park": 42,
"Delicate Arch": 18,
"Edinburgh Castle": 10,
"Eiffel Tower": 16,
"Externsteine": 23,
"Forth Bridge": 38,
"Gateway of India": 39,
"Golden Gate Bridge": 9,
"Grand Canyon": 8,
"Great Barrier Reef": 34,
"Great Wall of China": 46,
"Gullfoss Falls": 43,
"Haleakala National Park": 0,
"Hanging Temple": 32,
"Kantanagar Temple": 12,
"Ljubljana Castle": 2,
"London Olympic Stadium": 5,
"Machu Picchu": 41,
"Matterhorn": 20,
"Monumento a la Revolucion": 35,
"Moscow Raceway": 22,
"Mount Rainier National Park": 1,
"Mount Rushmore National Memorial": 11,
"Niagara Falls": 6,
"Petronas Towers": 29,
"Pont du Gard": 26,
"Prague Astronomical Clock": 47,
"Seattle Japanese Garden": 27,
"Soreq Cave": 24,
"Stockholm City Hall": 40,
"Stonehenge": 7,
"Sydney Harbour Bridge": 28,
"Sydney Opera House": 33,
"Taj Mahal": 21,
"Temple of Heaven": 45,
"Temple of Olympian Zeus": 49,
"Terminal Tower": 14,
"Trevi Fountain": 44,
"Vienna City Hall": 19,
"Washington Monument": 31,
"Whitby Abbey": 48,
"Wroclaws Dwarves": 4,
"Yellowstone National Park": 13
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"problem_type": "single_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.47.1"
}