Hugging Face's logo

YvesX
/
not-sea-otter

Image Classification
PyTorch
yolo
Model card Files Community
not-sea-otter
File size: 2,852 Bytes 
22e075e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
---
license: agpl-3.0
base_model:
- Ultralytics/YOLO11
tags:
- image-classification
- pytorch
- yolo
---

# Not Sea Otter

Yes, sea otters are incredibly CUTE, but you might not actually know what a sea otter looks like.

Can you distinguish between a sea otter, seal, sea dog, sea lion, walrus, manatee, and dugong?

Can you recognize the differences between sea otters, groundhogs, beavers, nutrias, capybaras, mongooses, and martens' face?

Can you tell apart a sea otter from a river otter? They are not the same!

This model was fine-tuned on YOLO11 in order to classify these animals often confused with sea otters.

The model was trained using thousands of animal images, with 80% for training and 20% for validation. It achieves an precision of 60% - 90% on the validation set, varying by animal.

## usage

```python
from huggingface_hub import hf_hub_download
from ultralytics import YOLO
from PIL import Image

model_path = hf_hub_download(repo_id="YvesX/not-sea-otter", filename="model.pt")

model = YOLO(model_path)

image_path = "/path/to/image.jpg"
image = Image.open(image_path).convert("RGB")

results = model(image)

if results[0].probs:
    top_class_index = results[0].probs.top1
    top_confidence = results[0].probs.top1conf.item()
    print(f"This is a {model.names[top_class_index]}! (with confidence: {top_confidence:.2f})")

```

## label

| id | label    | 中文   | English     | actually       | sample photos |
|----|----------|--------|-------------|----------------|---------------|
| 0  | capybara | 水豚   | Capybara    | Hydrochoerus   | 1851          |
| 1  | diao     | 貂     | Marten      | Martes         | 2669          |
| 2  | haibao   | 海豹   | Seal        | Phocidae       | 4397          |
| 3  | haigou   | 海狗   | Sea Dog     | Callorhinus    | 134           |
| 4  | hainiu   | 海牛   | Manatee     | Trichechidae   | 1118          |
| 5  | haishi   | 海狮   | Sea Lion    | Eumetopias     | 893           |
| 6  | haita    | 海獭   | Sea Otter   | Enhydra        | 1552          |
| 7  | haixiang | 海象   | Walrus      | Odobenidae     | 198           |
| 8  | heli     | 河狸   | Beaver      | Castor         | 4673          |
| 9  | helishu  | 河狸鼠 | Nutria      | Myocastor      | 3963          |
| 10 | meng     | 獴     | Mongoose    | Herpestidae    | 2083          |
| \* |          |        |             |                |               |
| 12 | rugen    | 儒艮   | Dugong      | Dugongidae     | 223           |
| 13 | shuita   | 水獭   | River Otter | Lutra          | 904           |
| 14 | tuboshu  | 土拨鼠 | Groundhog   | Marmota        | 3097          |


\* In Chinese, the pronunciation of "海豹" (seal) is identical to "海报" (poster). For humor somehow, the model also includes a `poster` label. ~~Movie poster recognition is quite accurate.~~