Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,60 @@
|
|
| 1 |
-
---
|
| 2 |
-
license:
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
|
| 4 |
+
tags:
|
| 5 |
+
- openclip
|
| 6 |
+
- vision-language
|
| 7 |
+
- image-classification
|
| 8 |
+
- zero-shot
|
| 9 |
+
- transfer-learning
|
| 10 |
+
library_name: timm
|
| 11 |
+
datasets:
|
| 12 |
+
- CUB_200_2011
|
| 13 |
+
- DTD
|
| 14 |
+
- FungiCLEF_2022
|
| 15 |
+
model-index:
|
| 16 |
+
- name: OpenCLIP - дообученные чекпоинты
|
| 17 |
+
results: []
|
| 18 |
+
|
| 19 |
+
---
|
| 20 |
+
|
| 21 |
+
# OpenCLIP: дообученные чекпоинты для различных доменов
|
| 22 |
+
|
| 23 |
+
Этот репозиторий содержит несколько чекпоинтов модели OpenCLIP, дообученных на различных датасетах для решения задач мультиклассовой и zero-shot классификации изображений.
|
| 24 |
+
|
| 25 |
+
## Описание модели
|
| 26 |
+
|
| 27 |
+
- **Базовая архитектура:** OpenCLIP (ViT backbone)
|
| 28 |
+
- **Дообучение:** Каждый чекпоинт соответствует разным экспериментам
|
| 29 |
+
- **Поддерживаемые задачи:** Классификация изображений
|
| 30 |
+
|
| 31 |
+
## Доступные чекпоинты
|
| 32 |
+
|
| 33 |
+
|
| 34 |
+
- `default_finetune_openclip_v1_epoch4_h.402...`:
|
| 35 |
+
Стандартное дообучение, OpenCLIP v1, 4 эпохи, best gzsl на всех классах - 0.382
|
| 36 |
+
|
| 37 |
+
- `default_finetune_openclip_v1_epoch1_uns0.38...`:
|
| 38 |
+
Стандартное дообучение, OpenCLIP, 1 эпоха, "uns0.38" — best тончность на unseen_classes - 0.44.
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
- `finetuned_openclip_v3_epoch1_h0.382...`:
|
| 42 |
+
Калибрация bias, OpenCLIP, 1 эпоха, "h0.382" — best gzsl на всех классах - 0.382
|
| 43 |
+
- `finetuned_openclip_v3_epoch1_uns0.44...`:
|
| 44 |
+
Калибрация bias, OpenCLIP, 1 эпоха, "uns0.44" — best тончность на unseen_classes - 0.44.
|
| 45 |
+
|
| 46 |
+
Каждый чекпоинт сохранён в формате PyTorch `.pickle` (~605 МБ) и требует наличия [Git LFS](https://git-lfs.com/).
|
| 47 |
+
|
| 48 |
+
## Использование
|
| 49 |
+
|
| 50 |
+
Загрузка чекпоинта в PyTorch:
|
| 51 |
+
```python
|
| 52 |
+
import torch
|
| 53 |
+
|
| 54 |
+
model = torch.load("finetuned_openclip_v3_epoch1_h0.3820202...pickle", map_location="cpu")
|
| 55 |
+
# Если используете официальный OpenCLIP:
|
| 56 |
+
# from open_clip import create_model_and_transforms
|
| 57 |
+
# model, _, _ = create_model_and_transforms('ViT-B/32', pretrained='laion2b_s34b_b79k')
|
| 58 |
+
|
| 59 |
+
|
| 60 |
+
```
|