Update README.md

README.md

@@ -3,7 +3,19 @@ license: cc-by-nc-sa-4.0
 ---
 This model is an object detection model trained with tensorflow object detection API, published with the paper [Edge Artificial Intelligence for real-time automatic quantification of filariasis in mobile microscopy](https://www.medrxiv.org/content/10.1101/2023.08.02.23293538v1)
 
-Model description:
-Developed by: Spotlab
-Model type: SSD mobilenet v2
-License: cc-by-nc-sa-4.0
+- Model description:
+- Developed by: Spotlab
+- Model type: SSD mobilenet v2
+- Classes: microfilaria
+- Datasets:
+  - Training set: 700 field of view images (100 magnification) from 85 samples with 1965 microfilarias
+  - Validation set: 173 field of view images (100 magnification) from 30 samples with 328 microfilarias
+  - Test set: 453 field of view images (100 magnification) from 30 samples with 328 microfilarias
+
+- Results:
+  - On validation set: 88.17% precision, 91.62% recall, and 89.85% f1 score.
+  - On test set: 94.14% precision, 91.90% recall, and 93.01% f1 score.
+Example detections
+  ![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/6509bcfc7e0d56c2717248be/t0rnWqTt69hbG4zPwimcu.jpeg)
+  ![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/6509bcfc7e0d56c2717248be/5i7H_eo_z7SFLAFrlL8-e.jpeg)
+