Initial model

README.md

@@ -1,3 +1,39 @@
----
-license: mit
----
+# Species Classification in Furry Art
+
+This model was trained on ~16k images (filtered from ~40k).
+I may train it on a larger dataset in the future.
+
+The model is capable of classifying about 86 species,
+including a few more common Pokemon, as well as some
+mythological and fictional species. The results may
+differ depending on the species, though the model is
+quite good with common ones such as foxes and wolves.
+
+## How to Use
+**Example using Pipelines (recommended):**
+```py
+from transformers import pipeline
+
+classify = pipeline('image-classification', model='ashduino101/furry-species-classification')
+
+scores = classify('your-image.png')
+# Do something with the scores
+```
+
+
+**Example using AutoModelForImageClassification:**
+```py
+import torch
+from PIL import Image
+from transformers import AutoImageProcessor, AutoModelForImageClassification
+
+image_processor = AutoImageProcessor.from_pretrained('ashduino101/furry-species-classification')
+inputs = image_processor(Image.open('your-image.png'), return_tensors='pt')
+
+model = AutoModelForImageClassification.from_pretrained("ashduino101/furry-species-classification")
+with torch.no_grad():
+    result = model(**inputs)
+
+# Do something with the result
+```
+

config.json

@@ -0,0 +1,200 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "alien",
+    "1": "angel",
+    "10": "centaur",
+    "11": "chicken",
+    "12": "chipmunk",
+    "13": "coyote",
+    "14": "cyborg",
+    "15": "cyclops",
+    "16": "deer",
+    "17": "demon",
+    "18": "dinosaur",
+    "19": "dolphin",
+    "2": "antelope",
+    "20": "donkey",
+    "21": "dragon",
+    "22": "eevee",
+    "23": "elf",
+    "24": "fairy",
+    "25": "feline",
+    "26": "fish",
+    "27": "flareon",
+    "28": "fox",
+    "29": "frog",
+    "3": "ape",
+    "30": "gazelle",
+    "31": "ghost",
+    "32": "giraffe",
+    "33": "glaceon",
+    "34": "goat",
+    "35": "gryphon",
+    "36": "hamster",
+    "37": "hedgehog",
+    "38": "horse",
+    "39": "human",
+    "4": "badger",
+    "40": "husky",
+    "41": "hyena",
+    "42": "imp",
+    "43": "jackal",
+    "44": "jaguar",
+    "45": "jolteon",
+    "46": "kangaroo",
+    "47": "kobold",
+    "48": "lemur",
+    "49": "lion",
+    "5": "bat_pony",
+    "50": "lizard",
+    "51": "lucario",
+    "52": "meerkat",
+    "53": "mollusk",
+    "54": "monkey",
+    "55": "monster",
+    "56": "mouse",
+    "57": "otter",
+    "58": "pegasus",
+    "59": "pikachu",
+    "6": "bear",
+    "60": "pony",
+    "61": "rabbit",
+    "62": "raccoon",
+    "63": "rat",
+    "64": "reindeer",
+    "65": "renamon",
+    "66": "robot",
+    "67": "rodent",
+    "68": "satyr",
+    "69": "sergal",
+    "7": "bird",
+    "70": "shark",
+    "71": "sheep",
+    "72": "sheepdog",
+    "73": "skunk",
+    "74": "snake",
+    "75": "sylveon",
+    "76": "tiger",
+    "77": "troll",
+    "78": "turtle",
+    "79": "umbreon",
+    "8": "braixen",
+    "80": "unicorn",
+    "81": "vaporeon",
+    "82": "weasel",
+    "83": "werewolf",
+    "84": "wolf",
+    "85": "yoshi",
+    "9": "cattle"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "alien": "0",
+    "angel": "1",
+    "antelope": "2",
+    "ape": "3",
+    "badger": "4",
+    "bat_pony": "5",
+    "bear": "6",
+    "bird": "7",
+    "braixen": "8",
+    "cattle": "9",
+    "centaur": "10",
+    "chicken": "11",
+    "chipmunk": "12",
+    "coyote": "13",
+    "cyborg": "14",
+    "cyclops": "15",
+    "deer": "16",
+    "demon": "17",
+    "dinosaur": "18",
+    "dolphin": "19",
+    "donkey": "20",
+    "dragon": "21",
+    "eevee": "22",
+    "elf": "23",
+    "fairy": "24",
+    "feline": "25",
+    "fish": "26",
+    "flareon": "27",
+    "fox": "28",
+    "frog": "29",
+    "gazelle": "30",
+    "ghost": "31",
+    "giraffe": "32",
+    "glaceon": "33",
+    "goat": "34",
+    "gryphon": "35",
+    "hamster": "36",
+    "hedgehog": "37",
+    "horse": "38",
+    "human": "39",
+    "husky": "40",
+    "hyena": "41",
+    "imp": "42",
+    "jackal": "43",
+    "jaguar": "44",
+    "jolteon": "45",
+    "kangaroo": "46",
+    "kobold": "47",
+    "lemur": "48",
+    "lion": "49",
+    "lizard": "50",
+    "lucario": "51",
+    "meerkat": "52",
+    "mollusk": "53",
+    "monkey": "54",
+    "monster": "55",
+    "mouse": "56",
+    "otter": "57",
+    "pegasus": "58",
+    "pikachu": "59",
+    "pony": "60",
+    "rabbit": "61",
+    "raccoon": "62",
+    "rat": "63",
+    "reindeer": "64",
+    "renamon": "65",
+    "robot": "66",
+    "rodent": "67",
+    "satyr": "68",
+    "sergal": "69",
+    "shark": "70",
+    "sheep": "71",
+    "sheepdog": "72",
+    "skunk": "73",
+    "snake": "74",
+    "sylveon": "75",
+    "tiger": "76",
+    "troll": "77",
+    "turtle": "78",
+    "umbreon": "79",
+    "unicorn": "80",
+    "vaporeon": "81",
+    "weasel": "82",
+    "werewolf": "83",
+    "wolf": "84",
+    "yoshi": "85"
+  },
+  "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.28.1"
+}

preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTFeatureExtractor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

pytorch_model.bin

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