Update README.md

README.md

@@ -13,11 +13,12 @@ license: cc-by-nc-4.0
 ## Model Details
 - **Model Type:** Danish Fungi Classification 
 - **Model Stats:**
-  - Params (M): ??
+  - Params (M): 50.2M
   - Image size: 299 x 299
 - **Papers:**
-- **Original:** ??
-- **Train Dataset:** DF20 --> https://sites.google.com/view/danish-fungi-dataset
+  - **Original 1:**  Aggregated Residual Transformations for Deep Neural Networks --> https://arxiv.org/pdf/1611.05431v2
+  - **Original 2:**  Squeeze-and-Excitation Networks --> https://arxiv.org/pdf/1709.01507v4
+  - **Train Dataset:** DF20 --> https://github.com/BohemianVRA/DanishFungiDataset/
 
 ## Model Usage
 ### Image Embeddings
@@ -40,18 +41,17 @@ output = model(train_transforms(img).unsqueeze(0))  # output is (batch_size, num
 ## 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,5 +59,4 @@ output = model(train_transforms(img).unsqueeze(0))  # output is (batch_size, num
   year={2022},
   publisher={Multidisciplinary Digital Publishing Institute}
 }
-```
-
+```