Update README.md

README.md

@@ -13,11 +13,11 @@ license: cc-by-nc-4.0
 ## Model Details
 - **Model Type:** Danish Fungi Classification 
 - **Model Stats:**
-  - Params (M): ??
+  - Params (M): 22.2M
   - Image size: 299 x 299
 - **Papers:**
-- **Original:** ??
-- **Train Dataset:** DF20M --> https://sites.google.com/view/danish-fungi-dataset
+  - **Original:** Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning --> https://arxiv.org/pdf/1602.07261
+  - **Train Dataset:** DF20M --> https://github.com/BohemianVRA/DanishFungiDataset/
 
 ## Model Usage
 ### Image Embeddings
@@ -36,22 +36,20 @@ img = Image.open(PATH_TO_YOUR_IMAGE)
 output = model(train_transforms(img).unsqueeze(0))  # output is (batch_size, num_features) shaped tensor
 ```
 
-
 ## Citation 
 ```bibtex
 @InProceedings{Picek_2022_WACV,
-    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},
+    author    = {Picek, Lukas and Sulc, Milan and Matas, Jiri and Jeppesen, Thomas S. and Heilmann-Clausen, Jacob and L{e}ss{\o}e, Thomas and Fr{\o}slev, Tobias},
     title     = {Danish Fungi 2020 - Not Just Another Image Recognition Dataset},
     booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
     month     = {January},
     year      = {2022},
     pages     = {1525-1535}
 }
-```
-```bibtex
+
 @article{picek2022automatic,
   title={Automatic Fungi Recognition: Deep Learning Meets Mycology},
-  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},
+  author={Picek, Lukas and Sulc, Milan and Matas, Jiri and Heilmann-Clausen, Jacob and Jeppesen, Thomas S and Lind, Emil},
   journal={Sensors},
   volume={22},
   number={2},
@@ -59,4 +57,4 @@ output = model(train_transforms(img).unsqueeze(0))  # output is (batch_size, num
   year={2022},
   publisher={Multidisciplinary Digital Publishing Institute}
 }
-```
+```