MHanzl commited on
Commit
5977c59
1 Parent(s): e914a6f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -4
README.md CHANGED
@@ -28,13 +28,33 @@ from PIL import Image
28
  from urllib.request import urlopen
29
  model = timm.create_model("hf-hub:BVRA/tf_efficientnet_b3.in1k_ft_df20m_384", pretrained=True)
30
  model = model.eval()
31
- train_transforms = T.Compose([T.Resize(384),
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
- https://openaccess.thecvf.com/content/WACV2022/papers/Picek_Danish_Fungi_2020_-_Not_Just_Another_Image_Recognition_Dataset_WACV_2022_paper.pdf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
28
  from urllib.request import urlopen
29
  model = timm.create_model("hf-hub:BVRA/tf_efficientnet_b3.in1k_ft_df20m_384", pretrained=True)
30
  model = model.eval()
31
+ train_transforms = T.Compose([T.Resize((384, 384)),
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
  ```
37
 
38
+ ## Citation
39
+ ```bibtex
40
+ @InProceedings{Picek_2022_WACV,
41
+ 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},
42
+ title = {Danish Fungi 2020 - Not Just Another Image Recognition Dataset},
43
+ booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
44
+ month = {January},
45
+ year = {2022},
46
+ pages = {1525-1535}
47
+ }
48
+ ```
49
+ ```bibtex
50
+ @article{picek2022automatic,
51
+ title={Automatic Fungi Recognition: Deep Learning Meets Mycology},
52
+ 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},
53
+ journal={Sensors},
54
+ volume={22},
55
+ number={2},
56
+ pages={633},
57
+ year={2022},
58
+ publisher={Multidisciplinary Digital Publishing Institute}
59
+ }
60
+ ```