nielsr HF Staff commited on
Commit
f6640a2
·
verified ·
1 Parent(s): 198c6ee

Add pipeline tag and link to Github repository

Browse files

This 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.

Files changed (1) hide show
  1. README.md +16 -6
README.md CHANGED
@@ -1,13 +1,14 @@
1
-
2
  ---
 
 
3
  tags:
4
  - image-classification
5
  - ecology
6
  - fungi
7
  - FGVC
8
- library_name: FungiTastic Dataset
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
- - **Original:** --> ???
20
- - **Train Dataset:** FungiTastic --> https://arxiv.org/pdf/2408.13632
 
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
+ ```