Add pipeline tag and link to Github repository
Browse filesThis PR improves the model card by:
- Adding the `pipeline_tag: image-classification` to the metadata, making the model discoverable via the Hugging Face model index.
- Adding the correct `library_name` (timm).
- Adding a link to the Github repository.
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/tf_efficientnetv2_b3.in1k_ft_fungitastic-mini_224
|
12 |
|
13 |
## Model Details
|
@@ -16,8 +17,9 @@ license: cc-by-nc-4.0
|
|
16 |
- Params (M): 13.2
|
17 |
- Image size: 224 x 224
|
18 |
- **Papers:**
|
19 |
-
- **
|
20 |
-
- **
|
|
|
21 |
|
22 |
## Model Usage
|
23 |
### Image Embeddings
|
@@ -37,6 +39,14 @@ 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 +77,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/tf_efficientnetv2_b3.in1k_ft_fungitastic-mini_224
|
13 |
|
14 |
## Model Details
|
|
|
17 |
- Params (M): 13.2
|
18 |
- Image size: 224 x 224
|
19 |
- **Papers:**
|
20 |
+
- **Fine-tuned model:** [FungiTastic: A multi-modal dataset and benchmark for image categorization](https://huggingface.co/papers/2408.13632)
|
21 |
+
- **Base Model:** tf_efficientnetv2_b3.in21k_ft_in1k
|
22 |
+
- **Train Dataset:** FungiTastic --> https://arxiv.org/pdf/2408.13632
|
23 |
|
24 |
## Model Usage
|
25 |
### Image Embeddings
|
|
|
39 |
# output is a (1, num_features) shaped tensor
|
40 |
```
|
41 |
|
42 |
+
## Project Page
|
43 |
+
|
44 |
+
https://bohemianvra.github.io/FungiTastic/
|
45 |
+
|
46 |
+
## Code
|
47 |
+
|
48 |
+
https://github.com/BohemianVRA/FungiTastic
|
49 |
+
|
50 |
## Citation
|
51 |
```bibtex
|
52 |
@article{picek2024fungitastic,
|
|
|
77 |
year={2022},
|
78 |
publisher={Multidisciplinary Digital Publishing Institute}
|
79 |
}
|
80 |
+
```
|