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