MHanzl commited on
Commit
6c70474
1 Parent(s): 1860a28

Upload README.md with huggingface_hub

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