File size: 1,287 Bytes
0d1baf7
 
 
 
 
 
 
 
 
519a2bc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
---
license: mit
tags:
- image-classification
- pytorch
- furry
- 
---

# 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
```