phytoViT/config.json

{
  "architectures": [
    "ViTForImageClassification"
  ],
  "attention_probs_dropout_prob": 0.0,
  "encoder_stride": 16,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.0,
  "hidden_size": 768,
  "id2label": {
    "0": "Akashiwo",
    "1": "Alexandrium",
    "2": "Amylax_Gonyaulax_Protoceratium",
    "3": "Asterionellopsis",
    "4": "Asterionellopsis_chain",
    "5": "Asteromphalus",
    "6": "Bad_Beads",
    "7": "Bad_blurred",
    "8": "Bad_mixed_phyto",
    "9": "Bad_setae",
    "10": "Centric",
    "11": "Centric_fuzzy",
    "12": "Ceratium_divaricatum",
    "13": "Ceratium_furca",
    "14": "Ceratium_lineatum",
    "15": "Chaetoceros",
    "16": "Ciliate_cutoff",
    "17": "Ciliate_large",
    "18": "Ciliate_large_2",
    "19": "Ciliate_other_morpho_1",
    "20": "Clusterflagellate_morpho_1",
    "21": "Clusterflagellate_morpho_2",
    "22": "Corethron",
    "23": "Cryptophyte",
    "24": "Cylindrotheca_Nitzschia",
    "25": "Detonula_Cerataulina_Lauderia",
    "26": "Detritus",
    "27": "Detritus_infection",
    "28": "Dictyocha",
    "29": "Dinoflagellate_morpho_1",
    "30": "Dinoflagellate_morpho_2",
    "31": "Dinoflagellate_morpho_3",
    "32": "Dinophysis",
    "33": "Ditylum",
    "34": "Entomoneis",
    "35": "Eucampia",
    "36": "Euglenoid",
    "37": "Flagellate_morpho_1",
    "38": "Flagellate_morpho_2",
    "39": "Flagellate_morpho_3",
    "40": "Flagellate_nano_1",
    "41": "Flagellate_nano_2",
    "42": "Fragilariopsis",
    "43": "Guinardia_Dactyliosolen",
    "44": "Gymnodinium",
    "45": "Gyrodinium",
    "46": "Gyrosigma",
    "47": "Haptophyte_prymnesium",
    "48": "Hemiaulus",
    "49": "Hemiselmis",
    "50": "Heterocapsa_morpho_1",
    "51": "Heterocapsa_morpho_2",
    "52": "Heterosigma_akashiwo",
    "53": "Laboea",
    "54": "Leptocylindrus",
    "55": "Lingulodinium",
    "56": "Margalefidinium",
    "57": "Mesodinium",
    "58": "Nano_cluster",
    "59": "Nano_p_white",
    "60": "Pennate_med",
    "61": "Pennate_morpho_1",
    "62": "Pennate_short",
    "63": "Pennate_wide",
    "64": "Peridinium",
    "65": "Phaeocystis",
    "66": "Pleurosigma",
    "67": "Polykrikos",
    "68": "Proboscia",
    "69": "Prorocentrum_narrow",
    "70": "Prorocentrum_wide",
    "71": "Pseudo-nitzschia",
    "72": "Pyramimonas",
    "73": "Rhizosolenia",
    "74": "Scrippsiella",
    "75": "Skeleonema",
    "76": "Skeletonema",
    "77": "Spiky_pacman_circular",
    "78": "Stombidinium_morpho_1",
    "79": "Strombidium_morpho_2",
    "80": "Thalassionema",
    "81": "Thalassiosira",
    "82": "Tiarina",
    "83": "Tintinnid",
    "84": "Tontonia",
    "85": "Torodinium",
    "86": "Tropidoneis",
    "87": "Unknown_morpho_1",
    "88": "Vicicitus"
  },
  "image_size": 224,
  "initializer_range": 0.02,
  "intermediate_size": 3072,
  "label2id": {
    "Akashiwo": 0,
    "Alexandrium": 1,
    "Amylax_Gonyaulax_Protoceratium": 2,
    "Asterionellopsis": 3,
    "Asterionellopsis_chain": 4,
    "Asteromphalus": 5,
    "Bad_Beads": 6,
    "Bad_blurred": 7,
    "Bad_mixed_phyto": 8,
    "Bad_setae": 9,
    "Centric": 10,
    "Centric_fuzzy": 11,
    "Ceratium_divaricatum": 12,
    "Ceratium_furca": 13,
    "Ceratium_lineatum": 14,
    "Chaetoceros": 15,
    "Ciliate_cutoff": 16,
    "Ciliate_large": 17,
    "Ciliate_large_2": 18,
    "Ciliate_other_morpho_1": 19,
    "Clusterflagellate_morpho_1": 20,
    "Clusterflagellate_morpho_2": 21,
    "Corethron": 22,
    "Cryptophyte": 23,
    "Cylindrotheca_Nitzschia": 24,
    "Detonula_Cerataulina_Lauderia": 25,
    "Detritus": 26,
    "Detritus_infection": 27,
    "Dictyocha": 28,
    "Dinoflagellate_morpho_1": 29,
    "Dinoflagellate_morpho_2": 30,
    "Dinoflagellate_morpho_3": 31,
    "Dinophysis": 32,
    "Ditylum": 33,
    "Entomoneis": 34,
    "Eucampia": 35,
    "Euglenoid": 36,
    "Flagellate_morpho_1": 37,
    "Flagellate_morpho_2": 38,
    "Flagellate_morpho_3": 39,
    "Flagellate_nano_1": 40,
    "Flagellate_nano_2": 41,
    "Fragilariopsis": 42,
    "Guinardia_Dactyliosolen": 43,
    "Gymnodinium": 44,
    "Gyrodinium": 45,
    "Gyrosigma": 46,
    "Haptophyte_prymnesium": 47,
    "Hemiaulus": 48,
    "Hemiselmis": 49,
    "Heterocapsa_morpho_1": 50,
    "Heterocapsa_morpho_2": 51,
    "Heterosigma_akashiwo": 52,
    "Laboea": 53,
    "Leptocylindrus": 54,
    "Lingulodinium": 55,
    "Margalefidinium": 56,
    "Mesodinium": 57,
    "Nano_cluster": 58,
    "Nano_p_white": 59,
    "Pennate_med": 60,
    "Pennate_morpho_1": 61,
    "Pennate_short": 62,
    "Pennate_wide": 63,
    "Peridinium": 64,
    "Phaeocystis": 65,
    "Pleurosigma": 66,
    "Polykrikos": 67,
    "Proboscia": 68,
    "Prorocentrum_narrow": 69,
    "Prorocentrum_wide": 70,
    "Pseudo-nitzschia": 71,
    "Pyramimonas": 72,
    "Rhizosolenia": 73,
    "Scrippsiella": 74,
    "Skeleonema": 75,
    "Skeletonema": 76,
    "Spiky_pacman_circular": 77,
    "Stombidinium_morpho_1": 78,
    "Strombidium_morpho_2": 79,
    "Thalassionema": 80,
    "Thalassiosira": 81,
    "Tiarina": 82,
    "Tintinnid": 83,
    "Tontonia": 84,
    "Torodinium": 85,
    "Tropidoneis": 86,
    "Unknown_morpho_1": 87,
    "Vicicitus": 88
  },
  "layer_norm_eps": 1e-12,
  "model_type": "vit",
  "num_attention_heads": 12,
  "num_channels": 3,
  "num_hidden_layers": 12,
  "patch_size": 16,
  "pooler_act": "tanh",
  "pooler_output_size": 768,
  "problem_type": "single_label_classification",
  "qkv_bias": true,
  "torch_dtype": "float32",
  "transformers_version": "4.52.4"
}