Add first version

README.md

@@ -1,3 +1,9 @@
----
-license: gpl-3.0
----
+# Poké Model
+
+Poké Model is a Pokémon Classifier created to be used with [Pokédex AI](https://github.com/torresflo/Pokedex-AI). It is a fine-tuned model of google/vit-base-patch16-224 to classify Pokémon of the first generation.
+
+More information on how to generate and how to use the model can be found on this [dedicated repository](https://github.com/torresflo/Poke-Model).
+
+## License
+
+Distributed under the GNU General Public License v3.0. See [here](https://www.gnu.org/licenses/gpl-3.0.en.html) for more information.

config.json

@@ -0,0 +1,330 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "bulbasaur",
+    "1": "ivysaur",
+    "2": "venusaur",
+    "3": "charmander",
+    "4": "charmeleon",
+    "5": "charizard",
+    "6": "squirtle",
+    "7": "wartortle",
+    "8": "blastoise",
+    "9": "caterpie",
+    "10": "metapod",
+    "11": "butterfree",
+    "12": "weedle",
+    "13": "kakuna",
+    "14": "beedrill",
+    "15": "pidgey",
+    "16": "pidgeotto",
+    "17": "pidgeot",
+    "18": "rattata",
+    "19": "raticate",
+    "20": "spearow",
+    "21": "fearow",
+    "22": "ekans",
+    "23": "arbok",
+    "24": "pikachu",
+    "25": "raichu",
+    "26": "sandshrew",
+    "27": "sandslash",
+    "28": "nidoran-f",
+    "29": "nidorina",
+    "30": "nidoqueen",
+    "31": "nidoran-m",
+    "32": "nidorino",
+    "33": "nidoking",
+    "34": "clefairy",
+    "35": "clefable",
+    "36": "vulpix",
+    "37": "ninetales",
+    "38": "jigglypuff",
+    "39": "wigglytuff",
+    "40": "zubat",
+    "41": "golbat",
+    "42": "oddish",
+    "43": "gloom",
+    "44": "vileplume",
+    "45": "paras",
+    "46": "parasect",
+    "47": "venonat",
+    "48": "venomoth",
+    "49": "diglett",
+    "50": "dugtrio",
+    "51": "meowth",
+    "52": "persian",
+    "53": "psyduck",
+    "54": "golduck",
+    "55": "mankey",
+    "56": "primeape",
+    "57": "growlithe",
+    "58": "arcanine",
+    "59": "poliwag",
+    "60": "poliwhirl",
+    "61": "poliwrath",
+    "62": "abra",
+    "63": "kadabra",
+    "64": "alakazam",
+    "65": "machop",
+    "66": "machoke",
+    "67": "machamp",
+    "68": "bellsprout",
+    "69": "weepinbell",
+    "70": "victreebel",
+    "71": "tentacool",
+    "72": "tentacruel",
+    "73": "geodude",
+    "74": "graveler",
+    "75": "golem",
+    "76": "ponyta",
+    "77": "rapidash",
+    "78": "slowpoke",
+    "79": "slowbro",
+    "80": "magnemite",
+    "81": "magneton",
+    "82": "farfetchd",
+    "83": "doduo",
+    "84": "dodrio",
+    "85": "seel",
+    "86": "dewgong",
+    "87": "grimer",
+    "88": "muk",
+    "89": "shellder",
+    "90": "cloyster",
+    "91": "gastly",
+    "92": "haunter",
+    "93": "gengar",
+    "94": "onix",
+    "95": "drowzee",
+    "96": "hypno",
+    "97": "krabby",
+    "98": "kingler",
+    "99": "voltorb",
+    "100": "electrode",
+    "101": "exeggcute",
+    "102": "exeggutor",
+    "103": "cubone",
+    "104": "marowak",
+    "105": "hitmonlee",
+    "106": "hitmonchan",
+    "107": "lickitung",
+    "108": "koffing",
+    "109": "weezing",
+    "110": "rhyhorn",
+    "111": "rhydon",
+    "112": "chansey",
+    "113": "tangela",
+    "114": "kangaskhan",
+    "115": "horsea",
+    "116": "seadra",
+    "117": "goldeen",
+    "118": "seaking",
+    "119": "staryu",
+    "120": "starmie",
+    "121": "mr-mime",
+    "122": "scyther",
+    "123": "jynx",
+    "124": "electabuzz",
+    "125": "magmar",
+    "126": "pinsir",
+    "127": "tauros",
+    "128": "magikarp",
+    "129": "gyarados",
+    "130": "lapras",
+    "131": "ditto",
+    "132": "eevee",
+    "133": "vaporeon",
+    "134": "jolteon",
+    "135": "flareon",
+    "136": "porygon",
+    "137": "omanyte",
+    "138": "omastar",
+    "139": "kabuto",
+    "140": "kabutops",
+    "141": "aerodactyl",
+    "142": "snorlax",
+    "143": "articuno",
+    "144": "zapdos",
+    "145": "moltres",
+    "146": "dratini",
+    "147": "dragonair",
+    "148": "dragonite",
+    "149": "mewtwo",
+    "150": "mew"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "abra": 62,
+    "aerodactyl": 141,
+    "alakazam": 64,
+    "arbok": 23,
+    "arcanine": 58,
+    "articuno": 143,
+    "beedrill": 14,
+    "bellsprout": 68,
+    "blastoise": 8,
+    "bulbasaur": 0,
+    "butterfree": 11,
+    "caterpie": 9,
+    "chansey": 112,
+    "charizard": 5,
+    "charmander": 3,
+    "charmeleon": 4,
+    "clefable": 35,
+    "clefairy": 34,
+    "cloyster": 90,
+    "cubone": 103,
+    "dewgong": 86,
+    "diglett": 49,
+    "ditto": 131,
+    "dodrio": 84,
+    "doduo": 83,
+    "dragonair": 147,
+    "dragonite": 148,
+    "dratini": 146,
+    "drowzee": 95,
+    "dugtrio": 50,
+    "eevee": 132,
+    "ekans": 22,
+    "electabuzz": 124,
+    "electrode": 100,
+    "exeggcute": 101,
+    "exeggutor": 102,
+    "farfetchd": 82,
+    "fearow": 21,
+    "flareon": 135,
+    "gastly": 91,
+    "gengar": 93,
+    "geodude": 73,
+    "gloom": 43,
+    "golbat": 41,
+    "goldeen": 117,
+    "golduck": 54,
+    "golem": 75,
+    "graveler": 74,
+    "grimer": 87,
+    "growlithe": 57,
+    "gyarados": 129,
+    "haunter": 92,
+    "hitmonchan": 106,
+    "hitmonlee": 105,
+    "horsea": 115,
+    "hypno": 96,
+    "ivysaur": 1,
+    "jigglypuff": 38,
+    "jolteon": 134,
+    "jynx": 123,
+    "kabuto": 139,
+    "kabutops": 140,
+    "kadabra": 63,
+    "kakuna": 13,
+    "kangaskhan": 114,
+    "kingler": 98,
+    "koffing": 108,
+    "krabby": 97,
+    "lapras": 130,
+    "lickitung": 107,
+    "machamp": 67,
+    "machoke": 66,
+    "machop": 65,
+    "magikarp": 128,
+    "magmar": 125,
+    "magnemite": 80,
+    "magneton": 81,
+    "mankey": 55,
+    "marowak": 104,
+    "meowth": 51,
+    "metapod": 10,
+    "mew": 150,
+    "mewtwo": 149,
+    "moltres": 145,
+    "mr-mime": 121,
+    "muk": 88,
+    "nidoking": 33,
+    "nidoqueen": 30,
+    "nidoran-f": 28,
+    "nidoran-m": 31,
+    "nidorina": 29,
+    "nidorino": 32,
+    "ninetales": 37,
+    "oddish": 42,
+    "omanyte": 137,
+    "omastar": 138,
+    "onix": 94,
+    "paras": 45,
+    "parasect": 46,
+    "persian": 52,
+    "pidgeot": 17,
+    "pidgeotto": 16,
+    "pidgey": 15,
+    "pikachu": 24,
+    "pinsir": 126,
+    "poliwag": 59,
+    "poliwhirl": 60,
+    "poliwrath": 61,
+    "ponyta": 76,
+    "porygon": 136,
+    "primeape": 56,
+    "psyduck": 53,
+    "raichu": 25,
+    "rapidash": 77,
+    "raticate": 19,
+    "rattata": 18,
+    "rhydon": 111,
+    "rhyhorn": 110,
+    "sandshrew": 26,
+    "sandslash": 27,
+    "scyther": 122,
+    "seadra": 116,
+    "seaking": 118,
+    "seel": 85,
+    "shellder": 89,
+    "slowbro": 79,
+    "slowpoke": 78,
+    "snorlax": 142,
+    "spearow": 20,
+    "squirtle": 6,
+    "starmie": 120,
+    "staryu": 119,
+    "tangela": 113,
+    "tauros": 127,
+    "tentacool": 71,
+    "tentacruel": 72,
+    "vaporeon": 133,
+    "venomoth": 48,
+    "venonat": 47,
+    "venusaur": 2,
+    "victreebel": 70,
+    "vileplume": 44,
+    "voltorb": 99,
+    "vulpix": 36,
+    "wartortle": 7,
+    "weedle": 12,
+    "weepinbell": 69,
+    "weezing": 109,
+    "wigglytuff": 39,
+    "zapdos": 144,
+    "zubat": 40
+  },
+  "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.25.1"
+}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d2929fbc8787b6ec07f9d2dcafa6d39aec7fdced704654792999929621ca402e
+size 343727021