Instructions to use SpotLab/microfilariae_detection with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SpotLab/microfilariae_detection with Transformers:
# Load model directly from transformers import SSDForObjectDetection model = SSDForObjectDetection.from_pretrained("SpotLab/microfilariae_detection", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| 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) | |
| - Developed by: Spotlab | |
| - Model type: SSD mobilenet v2 | |
| - Classes: Microfilaria | |
| - Model input: image resized to 640 and normalized to with mean=127.5 and std = 127.5. | |
| - 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 | |
| - Performance: | |
| - 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 | |
|  | |
|  | |
| The model is trained with the square inside the field of view instead with the full field of view. To ensure the model performace, use use this scale. | |
| You can create your own android app to run this model following this tutorial: [TensorFlow Lite Object Detection Android Demo | |
| ](https://github.com/tensorflow/examples/tree/master/lite/examples/object_detection/android) | |