picekl commited on
Commit
3827ecd
1 Parent(s): 30c1609

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +62 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - image-classification
5
+ - ecology
6
+ - fungi
7
+ - FGVC
8
+ library_name: DanishFungi
9
+ license: cc-by-nc-4.0
10
+ ---
11
+ # Model card for BVRA/resnet18.in1k_ft_fungitastic-mini_224
12
+
13
+ ## Model Details
14
+ - **Model Type:** Danish Fungi Classification
15
+ - **Model Stats:**
16
+ - Params (M): 11.3
17
+ - Image size: 224 x 224
18
+ - **Papers:**
19
+ - **Original:** ??
20
+ - **Train Dataset:** FungiTastic-Mini --> https://sites.google.com/view/danish-fungi-dataset
21
+
22
+ ## Model Usage
23
+ ### Image Embeddings
24
+ ```python
25
+ import timm
26
+ import torch
27
+ import torchvision.transforms as T
28
+ from PIL import Image
29
+ from urllib.request import urlopen
30
+ model = timm.create_model("hf-hub:BVRA/resnet18.in1k_ft_fungitastic-mini_224", pretrained=True)
31
+ model = model.eval()
32
+ train_transforms = T.Compose([T.Resize((224, 224)),
33
+ T.ToTensor(),
34
+ T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])
35
+ img = Image.open(PATH_TO_YOUR_IMAGE)
36
+ output = model(train_transforms(img).unsqueeze(0))
37
+ # output is a (1, num_features) shaped tensor
38
+ ```
39
+
40
+ ## Citation
41
+ ```bibtex
42
+ @InProceedings{Picek_2022_WACV,
43
+ author = {Picek, Luk'a {s} and {S}ulc, Milan and Matas, Ji {r}{'\i} and Jeppesen, Thomas S. and Heilmann-Clausen, Jacob and L{e}ss{\o}e, Thomas and Fr{\o}slev, Tobias},
44
+ title = {Danish Fungi 2020 - Not Just Another Image Recognition Dataset},
45
+ booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
46
+ month = {January},
47
+ year = {2022},
48
+ pages = {1525-1535}
49
+ }
50
+ ```
51
+ ```bibtex
52
+ @article{picek2022automatic,
53
+ title={Automatic Fungi Recognition: Deep Learning Meets Mycology},
54
+ author={Picek, Luk{'a}{ {s}} and { {S}}ulc, Milan and Matas, Ji{ {r}}{'\i} and Heilmann-Clausen, Jacob and Jeppesen, Thomas S and Lind, Emil},
55
+ journal={Sensors},
56
+ volume={22},
57
+ number={2},
58
+ pages={633},
59
+ year={2022},
60
+ publisher={Multidisciplinary Digital Publishing Institute}
61
+ }
62
+ ```