{ "_name_or_path": "microsoft/swinv2-large-patch4-window12to24-192to384-22kto1k-ft", "architectures": [ "Swinv2ForImageClassification" ], "attention_probs_dropout_prob": 0.0, "depths": [ 2, 2, 18, 2 ], "drop_path_rate": 0.1, "embed_dim": 192, "encoder_stride": 32, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 1536, "id2label": { "0": "Actinomycetes_Mycolata", "1": "Anaerobic Type B", "2": "Anaerobic suflur bacteria", "3": "Beggiatoa", "4": "Flexibacter", "5": "Fungi", "6": "GAO", "7": "Haliscomenobacter", "8": "Hyphomicrobium", "9": "Microscrilla", "10": "Microthrix", "11": "Nitrosomonas", "12": "Nostocoida limicola", "13": "PAO", "14": "Sphaerotilus-type 1701", "15": "Spirilla", "16": "Spirochaetes", "17": "Tetrads", "18": "Thiothrix", "19": "Type 0041_0675", "20": "Type 0092", "21": "Type 021N", "22": "Type 0411", "23": "Type 0581_Chloroflexi", "24": "Type 0914_0803", "25": "Type 0961", "26": "Type 1863", "27": "Zoogloea", "28": "bristleworm", "29": "crustacean", "30": "dead filament", "31": "elevated polysaccharide", "32": "ferric iron", "33": "fibrous material", "34": "filament impact on floc structure high", "35": "filament impact on floc structure low", "36": "filament impact on floc structure moderate", "37": "flagellate", "38": "floc open_diffuse", "39": "floc strong", "40": "floc weak", "41": "free swimming ciliate", "42": "gastrotrich", "43": "grease", "44": "inert", "45": "iron sulfide", "46": "irregular growth formations", "47": "naked amoebae", "48": "nematode", "49": "normal polysaccharide", "50": "oil", "51": "rotifer", "52": "stalked ciliate", "53": "testate amoebae", "54": "type 1851", "55": "water bear", "56": "yeast" }, "image_size": 384, "initializer_range": 0.02, "label2id": { "Actinomycetes_Mycolata": 0, "Anaerobic Type B": 1, "Anaerobic suflur bacteria": 2, "Beggiatoa": 3, "Flexibacter": 4, "Fungi": 5, "GAO": 6, "Haliscomenobacter": 7, "Hyphomicrobium": 8, "Microscrilla": 9, "Microthrix": 10, "Nitrosomonas": 11, "Nostocoida limicola": 12, "PAO": 13, "Sphaerotilus-type 1701": 14, "Spirilla": 15, "Spirochaetes": 16, "Tetrads": 17, "Thiothrix": 18, "Type 0041_0675": 19, "Type 0092": 20, "Type 021N": 21, "Type 0411": 22, "Type 0581_Chloroflexi": 23, "Type 0914_0803": 24, "Type 0961": 25, "Type 1863": 26, "Zoogloea": 27, "bristleworm": 28, "crustacean": 29, "dead filament": 30, "elevated polysaccharide": 31, "ferric iron": 32, "fibrous material": 33, "filament impact on floc structure high": 34, "filament impact on floc structure low": 35, "filament impact on floc structure moderate": 36, "flagellate": 37, "floc open_diffuse": 38, "floc strong": 39, "floc weak": 40, "free swimming ciliate": 41, "gastrotrich": 42, "grease": 43, "inert": 44, "iron sulfide": 45, "irregular growth formations": 46, "naked amoebae": 47, "nematode": 48, "normal polysaccharide": 49, "oil": 50, "rotifer": 51, "stalked ciliate": 52, "testate amoebae": 53, "type 1851": 54, "water bear": 55, "yeast": 56 }, "layer_norm_eps": 1e-05, "mlp_ratio": 4.0, "model_type": "swinv2", "num_channels": 3, "num_heads": [ 6, 12, 24, 48 ], "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.33.2", "use_absolute_embeddings": false, "window_size": 24 }