MHanzl commited on
Commit
eabedeb
1 Parent(s): 348ce37

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +28 -31
README.md CHANGED
@@ -1,3 +1,4 @@
 
1
  ---
2
  tags:
3
  - image-classification
@@ -6,38 +7,34 @@ tags:
6
  library_name: DanishFungi
7
  license: cc-by-nc-4.0
8
  ---
9
- # Model card for MHanzl/DF23M-mobilenetv2_100.ra_in1k_224
10
-
11
- A Swin-S image feature model. Superwisely pre-trained on animal re-identification datasets.
12
-
13
 
14
- ## Model Details
15
- - **Model Type:** Danish Fungi Classification
16
- - **Model Stats:**
17
- - Params (M): ??
18
- - Image size: [224, 224] x [224, 224]
19
- - **Papers:**
20
- - **Original:** ??
21
- - **Train Dataset:** DF23M --> https://sites.google.com/view/danish-fungi-dataset
22
 
23
- ## Model Usage
24
- ### Image Embeddings
25
- ```python
26
- import timm
27
- import torch
28
- import torchvision.transforms as T
29
- from PIL import Image
30
- from urllib.request import urlopen
31
- model = timm.create_model("hf-hub:MHanzl/DF23M-mobilenetv2_100.ra_in1k_224", pretrained=True)
32
- model = model.eval()
33
- train_transforms = T.Compose([T.Resize([224, 224]),
34
- T.ToTensor(),
35
- T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
36
- img = Image.open(PATH_TO_YOUR_IMAGE)
37
- output = model(train_transforms(img).unsqueeze(0)) # output is (batch_size, num_features) shaped tensor
38
- # output is a (1, num_features) shaped tensor
39
- ```
40
 
41
- ## Citation
42
 
43
-
 
1
+
2
  ---
3
  tags:
4
  - image-classification
 
7
  library_name: DanishFungi
8
  license: cc-by-nc-4.0
9
  ---
10
+ # Model card for MHanzl/DF23M-mobilenetv2_100.ra_in1k_224
 
 
 
11
 
12
+ ## Model Details
13
+ - **Model Type:** Danish Fungi Classification
14
+ - **Model Stats:**
15
+ - Params (M): ??
16
+ - Image size: [224, 224] x [224, 224]
17
+ - **Papers:**
18
+ - **Original:** ??
19
+ - **Train Dataset:** DF23M --> https://sites.google.com/view/danish-fungi-dataset
20
 
21
+ ## Model Usage
22
+ ### Image Embeddings
23
+ ```python
24
+ import timm
25
+ import torch
26
+ import torchvision.transforms as T
27
+ from PIL import Image
28
+ from urllib.request import urlopen
29
+ model = timm.create_model("hf-hub:MHanzl/DF23M-mobilenetv2_100.ra_in1k_224", pretrained=True)
30
+ model = model.eval()
31
+ train_transforms = T.Compose([T.Resize([224, 224]),
32
+ T.ToTensor(),
33
+ T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
34
+ img = Image.open(PATH_TO_YOUR_IMAGE)
35
+ output = model(train_transforms(img).unsqueeze(0)) # output is (batch_size, num_features) shaped tensor
36
+ # output is a (1, num_features) shaped tensor
37
+ ```
38
 
39
+ ## Citation
40