nielsr HF Staff commited on
Commit
ebe9b22
·
verified ·
1 Parent(s): 4f12440

Improve model card metadata and add project page link

Browse files

This PR improves the model card by:

- Adding the `pipeline_tag: image-classification` to correctly categorize the model.
- Updating the `library_name` to `timm` to reflect the model architecture.
- Adding a link to the project page for more details.
- Adding link to the original paper.

Files changed (1) hide show
  1. README.md +8 -5
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/convnext_base.in1k_ft_fungitastic-mini_224
12
 
13
  ## Model Details
@@ -16,9 +17,11 @@ license: cc-by-nc-4.0
16
  - Params (M): 87.8
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
24
  ```python
@@ -67,4 +70,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/convnext_base.in1k_ft_fungitastic-mini_224
13
 
14
  ## Model Details
 
17
  - Params (M): 87.8
18
  - Image size: 224 x 224
19
  - **Papers:**
20
+ - **Original:** [ConvNeXt](https://arxiv.org/abs/2201.03545)
21
  - **Train Dataset:** FungiTastic --> https://arxiv.org/pdf/2408.13632
22
 
23
+ Project page: https://bohemianvra.github.io/FungiTastic/
24
+
25
  ## Model Usage
26
  ### Image Embeddings
27
  ```python
 
70
  year={2022},
71
  publisher={Multidisciplinary Digital Publishing Institute}
72
  }
73
+ ```