{ | |
"_name_or_path": "microsoft/swinv2-small-patch4-window16-256", | |
"architectures": [ | |
"Swinv2ForImageClassification" | |
], | |
"attention_probs_dropout_prob": 0.0, | |
"depths": [ | |
2, | |
2, | |
18, | |
2 | |
], | |
"drop_path_rate": 0.1, | |
"embed_dim": 96, | |
"encoder_stride": 32, | |
"hidden_act": "gelu", | |
"hidden_dropout_prob": 0.0, | |
"hidden_size": 768, | |
"id2label": { | |
"0": "acanthite", | |
"1": "actinolite", | |
"2": "adamite", | |
"3": "aegirine", | |
"4": "afghanite", | |
"5": "agate", | |
"6": "alabandite", | |
"7": "albite", | |
"8": "almandine", | |
"9": "amethyst", | |
"10": "analcime", | |
"11": "anatase", | |
"12": "andalusite", | |
"13": "andradite", | |
"14": "anglesite", | |
"15": "anhydrite", | |
"16": "annabergite", | |
"17": "anorthite", | |
"18": "anorthoclase", | |
"19": "antimony", | |
"20": "apatite", | |
"21": "apophyllite", | |
"22": "aquamarine", | |
"23": "aragonite", | |
"24": "arfvedsonite", | |
"25": "arsenic", | |
"26": "arsenopyrite", | |
"27": "augite", | |
"28": "aurichalcite", | |
"29": "axinite", | |
"30": "azurite", | |
"31": "babingtonite", | |
"32": "barite", | |
"33": "bauxite", | |
"34": "benitoite", | |
"35": "beryl", | |
"36": "biotite", | |
"37": "bismuth", | |
"38": "bismuthinite", | |
"39": "bixbyite", | |
"40": "boehmite", | |
"41": "boleite", | |
"42": "boracite", | |
"43": "borax", | |
"44": "bornite", | |
"45": "boulangerite", | |
"46": "bournonite", | |
"47": "brochantite", | |
"48": "bromargyrite", | |
"49": "brookite", | |
"50": "brucite", | |
"51": "buergerite", | |
"52": "bustamite", | |
"53": "bytownite", | |
"54": "calaverite", | |
"55": "calcite", | |
"56": "calomel", | |
"57": "carletonite", | |
"58": "cassiterite", | |
"59": "celestine", | |
"60": "cerussite", | |
"61": "cervantite", | |
"62": "chabazite", | |
"63": "chalcanthite", | |
"64": "chalcedony", | |
"65": "chalcocite", | |
"66": "chalcopyrite", | |
"67": "chambersite", | |
"68": "chamosite", | |
"69": "chlorargyrite", | |
"70": "chlorite", | |
"71": "chondrodite", | |
"72": "chromite", | |
"73": "chrysoberyl", | |
"74": "chrysocolla", | |
"75": "cinnabar", | |
"76": "citrine", | |
"77": "clinochlore", | |
"78": "clinohumite", | |
"79": "clinozoisite", | |
"80": "cobaltite", | |
"81": "coesite", | |
"82": "colemanite", | |
"83": "columbite", | |
"84": "cookeite", | |
"85": "copper", | |
"86": "cordierite", | |
"87": "corundum", | |
"88": "covellite", | |
"89": "cristobalite", | |
"90": "crocoite", | |
"91": "cryolite", | |
"92": "cumengeite", | |
"93": "cuprite", | |
"94": "cyanotrichite", | |
"95": "danburite", | |
"96": "datolite", | |
"97": "diamond", | |
"98": "diaspore", | |
"99": "diopside", | |
"100": "dioptase", | |
"101": "dolomite", | |
"102": "dravite", | |
"103": "dumortierite", | |
"104": "edenite", | |
"105": "elbaite", | |
"106": "emerald", | |
"107": "enargite", | |
"108": "enstatite", | |
"109": "epidote", | |
"110": "epistilbite", | |
"111": "erythrite", | |
"112": "euclase", | |
"113": "ferberite", | |
"114": "fluorite", | |
"115": "franklinite", | |
"116": "galena", | |
"117": "garnet", | |
"118": "gaspeite", | |
"119": "gibbsite", | |
"120": "glauberite", | |
"121": "glaucophane", | |
"122": "gmelinite", | |
"123": "goethite", | |
"124": "gold", | |
"125": "goosecreekite", | |
"126": "graphite", | |
"127": "greenockite", | |
"128": "grossular", | |
"129": "gummite", | |
"130": "gypsum", | |
"131": "halite", | |
"132": "hastingsite", | |
"133": "hauyne", | |
"134": "hedenbergite", | |
"135": "hematite", | |
"136": "hemimorphite", | |
"137": "heulandite", | |
"138": "hornblende", | |
"139": "howlite", | |
"140": "huebnerite", | |
"141": "humite", | |
"142": "hydrozincite", | |
"143": "ice", | |
"144": "ilmenite", | |
"145": "inesite", | |
"146": "iodargyrite", | |
"147": "iron-nickel", | |
"148": "jadeite", | |
"149": "jamesonite", | |
"150": "jarosite", | |
"151": "jasper", | |
"152": "jeremejevite", | |
"153": "johannsenite", | |
"154": "kaolinite", | |
"155": "kernite", | |
"156": "krennerite", | |
"157": "kyanite", | |
"158": "labradorite", | |
"159": "laumontite", | |
"160": "lawsonite", | |
"161": "lazulite", | |
"162": "lazurite", | |
"163": "lead", | |
"164": "lepidolite", | |
"165": "leucite", | |
"166": "liddicoatite", | |
"167": "limonite", | |
"168": "linarite", | |
"169": "lithiophilite", | |
"170": "loellingite", | |
"171": "luzonite", | |
"172": "magnesiochromite", | |
"173": "magnesite", | |
"174": "magnetite", | |
"175": "malachite", | |
"176": "manganite", | |
"177": "manganoneptunite", | |
"178": "marcasite", | |
"179": "marialite", | |
"180": "meionite", | |
"181": "melanophlogite", | |
"182": "mercury", | |
"183": "mesolite", | |
"184": "metacinnabar", | |
"185": "metastibnite", | |
"186": "metavariscite", | |
"187": "miargyrite", | |
"188": "microcline", | |
"189": "millerite", | |
"190": "mimetite", | |
"191": "molybdenite", | |
"192": "monazite", | |
"193": "muscovite", | |
"194": "natrolite", | |
"195": "nepheline", | |
"196": "neptunite", | |
"197": "nickeline", | |
"198": "nitratine", | |
"199": "norbergite", | |
"200": "nosean", | |
"201": "oligoclase", | |
"202": "olivine", | |
"203": "olmiite", | |
"204": "opal", | |
"205": "orpiment", | |
"206": "orthoclase", | |
"207": "otavite", | |
"208": "paradamite", | |
"209": "pararealgar", | |
"210": "pargasite", | |
"211": "pearceite", | |
"212": "pectolite", | |
"213": "pentlandite", | |
"214": "pezzottaite", | |
"215": "phenakite", | |
"216": "phlogopite", | |
"217": "phosgenite", | |
"218": "plagioclase", | |
"219": "platinum", | |
"220": "polybasite", | |
"221": "powellite", | |
"222": "prehnite", | |
"223": "proustite", | |
"224": "psilomelane", | |
"225": "pumpellyite", | |
"226": "pyrargyrite", | |
"227": "pyrite", | |
"228": "pyrolusite", | |
"229": "pyromorphite", | |
"230": "pyrope", | |
"231": "pyrophyllite", | |
"232": "pyroxmangite", | |
"233": "pyrrhotite", | |
"234": "quartz", | |
"235": "rammelsbergite", | |
"236": "raspite", | |
"237": "realgar", | |
"238": "rheniite", | |
"239": "rhodochrosite", | |
"240": "rhodonite", | |
"241": "riebeckite", | |
"242": "romanechite", | |
"243": "rutile", | |
"244": "talc", | |
"245": "tantalite", | |
"246": "tellurium", | |
"247": "tennantite", | |
"248": "tephroite", | |
"249": "tetrahedrite", | |
"250": "thenardite", | |
"251": "thomsonite", | |
"252": "thorite", | |
"253": "tincalconite", | |
"254": "titanite", | |
"255": "topaz", | |
"256": "tourmaline", | |
"257": "tremolite", | |
"258": "tridymite", | |
"259": "triphylite", | |
"260": "turquoise", | |
"261": "ulexite", | |
"262": "uraninite", | |
"263": "uvarovite", | |
"264": "uvite", | |
"265": "vanadinite", | |
"266": "variscite", | |
"267": "vesuvianite", | |
"268": "vivianite", | |
"269": "water", | |
"270": "wavellite", | |
"271": "willemite", | |
"272": "witherite", | |
"273": "wolframite", | |
"274": "wollastonite", | |
"275": "wulfenite", | |
"276": "wurtzite", | |
"277": "xanthoconite", | |
"278": "xenotime", | |
"279": "zincite", | |
"280": "zircon", | |
"281": "zoisite" | |
}, | |
"image_size": 256, | |
"initializer_range": 0.02, | |
"label2id": { | |
"acanthite": 0, | |
"actinolite": 1, | |
"adamite": 2, | |
"aegirine": 3, | |
"afghanite": 4, | |
"agate": 5, | |
"alabandite": 6, | |
"albite": 7, | |
"almandine": 8, | |
"amethyst": 9, | |
"analcime": 10, | |
"anatase": 11, | |
"andalusite": 12, | |
"andradite": 13, | |
"anglesite": 14, | |
"anhydrite": 15, | |
"annabergite": 16, | |
"anorthite": 17, | |
"anorthoclase": 18, | |
"antimony": 19, | |
"apatite": 20, | |
"apophyllite": 21, | |
"aquamarine": 22, | |
"aragonite": 23, | |
"arfvedsonite": 24, | |
"arsenic": 25, | |
"arsenopyrite": 26, | |
"augite": 27, | |
"aurichalcite": 28, | |
"axinite": 29, | |
"azurite": 30, | |
"babingtonite": 31, | |
"barite": 32, | |
"bauxite": 33, | |
"benitoite": 34, | |
"beryl": 35, | |
"biotite": 36, | |
"bismuth": 37, | |
"bismuthinite": 38, | |
"bixbyite": 39, | |
"boehmite": 40, | |
"boleite": 41, | |
"boracite": 42, | |
"borax": 43, | |
"bornite": 44, | |
"boulangerite": 45, | |
"bournonite": 46, | |
"brochantite": 47, | |
"bromargyrite": 48, | |
"brookite": 49, | |
"brucite": 50, | |
"buergerite": 51, | |
"bustamite": 52, | |
"bytownite": 53, | |
"calaverite": 54, | |
"calcite": 55, | |
"calomel": 56, | |
"carletonite": 57, | |
"cassiterite": 58, | |
"celestine": 59, | |
"cerussite": 60, | |
"cervantite": 61, | |
"chabazite": 62, | |
"chalcanthite": 63, | |
"chalcedony": 64, | |
"chalcocite": 65, | |
"chalcopyrite": 66, | |
"chambersite": 67, | |
"chamosite": 68, | |
"chlorargyrite": 69, | |
"chlorite": 70, | |
"chondrodite": 71, | |
"chromite": 72, | |
"chrysoberyl": 73, | |
"chrysocolla": 74, | |
"cinnabar": 75, | |
"citrine": 76, | |
"clinochlore": 77, | |
"clinohumite": 78, | |
"clinozoisite": 79, | |
"cobaltite": 80, | |
"coesite": 81, | |
"colemanite": 82, | |
"columbite": 83, | |
"cookeite": 84, | |
"copper": 85, | |
"cordierite": 86, | |
"corundum": 87, | |
"covellite": 88, | |
"cristobalite": 89, | |
"crocoite": 90, | |
"cryolite": 91, | |
"cumengeite": 92, | |
"cuprite": 93, | |
"cyanotrichite": 94, | |
"danburite": 95, | |
"datolite": 96, | |
"diamond": 97, | |
"diaspore": 98, | |
"diopside": 99, | |
"dioptase": 100, | |
"dolomite": 101, | |
"dravite": 102, | |
"dumortierite": 103, | |
"edenite": 104, | |
"elbaite": 105, | |
"emerald": 106, | |
"enargite": 107, | |
"enstatite": 108, | |
"epidote": 109, | |
"epistilbite": 110, | |
"erythrite": 111, | |
"euclase": 112, | |
"ferberite": 113, | |
"fluorite": 114, | |
"franklinite": 115, | |
"galena": 116, | |
"garnet": 117, | |
"gaspeite": 118, | |
"gibbsite": 119, | |
"glauberite": 120, | |
"glaucophane": 121, | |
"gmelinite": 122, | |
"goethite": 123, | |
"gold": 124, | |
"goosecreekite": 125, | |
"graphite": 126, | |
"greenockite": 127, | |
"grossular": 128, | |
"gummite": 129, | |
"gypsum": 130, | |
"halite": 131, | |
"hastingsite": 132, | |
"hauyne": 133, | |
"hedenbergite": 134, | |
"hematite": 135, | |
"hemimorphite": 136, | |
"heulandite": 137, | |
"hornblende": 138, | |
"howlite": 139, | |
"huebnerite": 140, | |
"humite": 141, | |
"hydrozincite": 142, | |
"ice": 143, | |
"ilmenite": 144, | |
"inesite": 145, | |
"iodargyrite": 146, | |
"iron-nickel": 147, | |
"jadeite": 148, | |
"jamesonite": 149, | |
"jarosite": 150, | |
"jasper": 151, | |
"jeremejevite": 152, | |
"johannsenite": 153, | |
"kaolinite": 154, | |
"kernite": 155, | |
"krennerite": 156, | |
"kyanite": 157, | |
"labradorite": 158, | |
"laumontite": 159, | |
"lawsonite": 160, | |
"lazulite": 161, | |
"lazurite": 162, | |
"lead": 163, | |
"lepidolite": 164, | |
"leucite": 165, | |
"liddicoatite": 166, | |
"limonite": 167, | |
"linarite": 168, | |
"lithiophilite": 169, | |
"loellingite": 170, | |
"luzonite": 171, | |
"magnesiochromite": 172, | |
"magnesite": 173, | |
"magnetite": 174, | |
"malachite": 175, | |
"manganite": 176, | |
"manganoneptunite": 177, | |
"marcasite": 178, | |
"marialite": 179, | |
"meionite": 180, | |
"melanophlogite": 181, | |
"mercury": 182, | |
"mesolite": 183, | |
"metacinnabar": 184, | |
"metastibnite": 185, | |
"metavariscite": 186, | |
"miargyrite": 187, | |
"microcline": 188, | |
"millerite": 189, | |
"mimetite": 190, | |
"molybdenite": 191, | |
"monazite": 192, | |
"muscovite": 193, | |
"natrolite": 194, | |
"nepheline": 195, | |
"neptunite": 196, | |
"nickeline": 197, | |
"nitratine": 198, | |
"norbergite": 199, | |
"nosean": 200, | |
"oligoclase": 201, | |
"olivine": 202, | |
"olmiite": 203, | |
"opal": 204, | |
"orpiment": 205, | |
"orthoclase": 206, | |
"otavite": 207, | |
"paradamite": 208, | |
"pararealgar": 209, | |
"pargasite": 210, | |
"pearceite": 211, | |
"pectolite": 212, | |
"pentlandite": 213, | |
"pezzottaite": 214, | |
"phenakite": 215, | |
"phlogopite": 216, | |
"phosgenite": 217, | |
"plagioclase": 218, | |
"platinum": 219, | |
"polybasite": 220, | |
"powellite": 221, | |
"prehnite": 222, | |
"proustite": 223, | |
"psilomelane": 224, | |
"pumpellyite": 225, | |
"pyrargyrite": 226, | |
"pyrite": 227, | |
"pyrolusite": 228, | |
"pyromorphite": 229, | |
"pyrope": 230, | |
"pyrophyllite": 231, | |
"pyroxmangite": 232, | |
"pyrrhotite": 233, | |
"quartz": 234, | |
"rammelsbergite": 235, | |
"raspite": 236, | |
"realgar": 237, | |
"rheniite": 238, | |
"rhodochrosite": 239, | |
"rhodonite": 240, | |
"riebeckite": 241, | |
"romanechite": 242, | |
"rutile": 243, | |
"talc": 244, | |
"tantalite": 245, | |
"tellurium": 246, | |
"tennantite": 247, | |
"tephroite": 248, | |
"tetrahedrite": 249, | |
"thenardite": 250, | |
"thomsonite": 251, | |
"thorite": 252, | |
"tincalconite": 253, | |
"titanite": 254, | |
"topaz": 255, | |
"tourmaline": 256, | |
"tremolite": 257, | |
"tridymite": 258, | |
"triphylite": 259, | |
"turquoise": 260, | |
"ulexite": 261, | |
"uraninite": 262, | |
"uvarovite": 263, | |
"uvite": 264, | |
"vanadinite": 265, | |
"variscite": 266, | |
"vesuvianite": 267, | |
"vivianite": 268, | |
"water": 269, | |
"wavellite": 270, | |
"willemite": 271, | |
"witherite": 272, | |
"wolframite": 273, | |
"wollastonite": 274, | |
"wulfenite": 275, | |
"wurtzite": 276, | |
"xanthoconite": 277, | |
"xenotime": 278, | |
"zincite": 279, | |
"zircon": 280, | |
"zoisite": 281 | |
}, | |
"layer_norm_eps": 1e-05, | |
"mlp_ratio": 4.0, | |
"model_type": "swinv2", | |
"num_channels": 3, | |
"num_heads": [ | |
3, | |
6, | |
12, | |
24 | |
], | |
"num_layers": 4, | |
"patch_size": 4, | |
"path_norm": true, | |
"pretrained_window_sizes": [ | |
0, | |
0, | |
0, | |
0 | |
], | |
"problem_type": "single_label_classification", | |
"qkv_bias": true, | |
"torch_dtype": "float32", | |
"transformers_version": "4.31.0", | |
"use_absolute_embeddings": false, | |
"window_size": 16 | |
} | |