Improve model card: Add pipeline tag, project page link, and link to paper
#1
by
nielsr
HF Staff
- opened
README.md
CHANGED
@@ -1,13 +1,14 @@
|
|
1 |
-
|
2 |
---
|
|
|
|
|
3 |
tags:
|
4 |
- image-classification
|
5 |
- ecology
|
6 |
- fungi
|
7 |
- FGVC
|
8 |
-
|
9 |
-
license: cc-by-nc-4.0
|
10 |
---
|
|
|
11 |
# Model card for BVRA/swin_base_patch4_window7_224.in1k_ft_fungitastic-mini_224
|
12 |
|
13 |
## Model Details
|
@@ -16,8 +17,8 @@ license: cc-by-nc-4.0
|
|
16 |
- Params (M): 87.0
|
17 |
- Image size: 224 x 224
|
18 |
- **Papers:**
|
19 |
-
- **Original:**
|
20 |
-
- **Train Dataset:** FungiTastic
|
21 |
|
22 |
## Model Usage
|
23 |
### Image Embeddings
|
@@ -25,7 +26,7 @@ license: cc-by-nc-4.0
|
|
25 |
import timm
|
26 |
import torch
|
27 |
import torchvision.transforms as T
|
28 |
-
from PIL
|
29 |
from urllib.request import urlopen
|
30 |
model = timm.create_model("hf-hub:BVRA/swin_base_patch4_window7_224.in1k_ft_fungitastic-mini_224", pretrained=True)
|
31 |
model = model.eval()
|
@@ -37,6 +38,9 @@ output = model(train_transforms(img).unsqueeze(0))
|
|
37 |
# output is a (1, num_features) shaped tensor
|
38 |
```
|
39 |
|
|
|
|
|
|
|
40 |
## Citation
|
41 |
```bibtex
|
42 |
@article{picek2024fungitastic,
|
@@ -67,4 +71,4 @@ output = model(train_transforms(img).unsqueeze(0))
|
|
67 |
year={2022},
|
68 |
publisher={Multidisciplinary Digital Publishing Institute}
|
69 |
}
|
70 |
-
```
|
|
|
|
|
1 |
---
|
2 |
+
library_name: timm
|
3 |
+
license: cc-by-nc-4.0
|
4 |
tags:
|
5 |
- image-classification
|
6 |
- ecology
|
7 |
- fungi
|
8 |
- FGVC
|
9 |
+
pipeline_tag: image-classification
|
|
|
10 |
---
|
11 |
+
|
12 |
# Model card for BVRA/swin_base_patch4_window7_224.in1k_ft_fungitastic-mini_224
|
13 |
|
14 |
## Model Details
|
|
|
17 |
- Params (M): 87.0
|
18 |
- Image size: 224 x 224
|
19 |
- **Papers:**
|
20 |
+
- **Original:** [FungiTastic: A multi-modal dataset and benchmark for image categorization](https://huggingface.co/papers/2408.13632)
|
21 |
+
- **Train Dataset:** FungiTastic
|
22 |
|
23 |
## Model Usage
|
24 |
### Image Embeddings
|
|
|
26 |
import timm
|
27 |
import torch
|
28 |
import torchvision.transforms as T
|
29 |
+
from PIL Image
|
30 |
from urllib.request import urlopen
|
31 |
model = timm.create_model("hf-hub:BVRA/swin_base_patch4_window7_224.in1k_ft_fungitastic-mini_224", pretrained=True)
|
32 |
model = model.eval()
|
|
|
38 |
# output is a (1, num_features) shaped tensor
|
39 |
```
|
40 |
|
41 |
+
## Project Page
|
42 |
+
For more details about the FungiTastic dataset and benchmark, please refer to the project page: [https://bohemianvra.github.io/FungiTastic/](https://bohemianvra.github.io/FungiTastic/)
|
43 |
+
|
44 |
## Citation
|
45 |
```bibtex
|
46 |
@article{picek2024fungitastic,
|
|
|
71 |
year={2022},
|
72 |
publisher={Multidisciplinary Digital Publishing Institute}
|
73 |
}
|
74 |
+
```
|