Update README.md
Browse files
README.md
CHANGED
|
@@ -1,33 +1,77 @@
|
|
| 1 |
-
---
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
-
|
| 5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
-
#
|
| 8 |
|
| 9 |
-
|
| 10 |
|
| 11 |
-
|
| 12 |
-
1. Fill out this model card with more information (see the template below and the [documentation here](https://huggingface.co/docs/hub/model-repos))!
|
| 13 |
|
| 14 |
-
|
| 15 |
|
| 16 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
-
|
| 19 |
|
|
|
|
| 20 |
|
| 21 |
-
|
|
|
|
|
|
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
-
|
| 25 |
|
| 26 |
-
|
| 27 |
-
More information needed
|
| 28 |
|
| 29 |
-
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
|
| 32 |
-
|
| 33 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- image-classification
|
| 4 |
+
- timm
|
| 5 |
+
- transformers
|
| 6 |
+
- fastai
|
| 7 |
+
library_name: fastai
|
| 8 |
+
license: apache-2.0
|
| 9 |
+
datasets:
|
| 10 |
+
- imagenet-1k
|
| 11 |
+
- imagenet-22k
|
| 12 |
+
- iloncka/mosquito-species-classification-dataset
|
| 13 |
+
metrics:
|
| 14 |
+
- accuracy
|
| 15 |
+
base_model:
|
| 16 |
+
- timm/tiny_vit_21m_224.dist_in22k_ft_in1k
|
| 17 |
+
---
|
| 18 |
|
| 19 |
+
# Model Card for `culico-net-cls-v1`
|
| 20 |
|
| 21 |
+
`culico-net-cls-v1` - image classification model focused on identifying mosquito species. This model is a result of the `CulicidaeLab` project and was developed by fine-tuning the `tiny_vit_21m_224.dist_in22k_ft_in1k` model.
|
| 22 |
|
| 23 |
+
The `culico-net-cls-v1` is a TinyViT image classification model. It was pretrained on the large-scale ImageNet-22k dataset using distillation and then fine-tuned on the ImageNet-1k dataset by the original paper's authors. This foundational training has been further adapted for the specific task of mosquito species classification using a dedicated dataset.
|
|
|
|
| 24 |
|
| 25 |
+
**Model Details:**
|
| 26 |
|
| 27 |
+
* **Model Type:** Image classification / feature backbone
|
| 28 |
+
* **Model Stats:**
|
| 29 |
+
* Parameters (M): 21.2
|
| 30 |
+
* GMACs: 4.1
|
| 31 |
+
* Activations (M): 15.9
|
| 32 |
+
* Image size: 224 x 224
|
| 33 |
+
* **Papers:**
|
| 34 |
+
* TinyViT: Fast Pretraining Distillation for Small Vision Transformers: https://arxiv.org/abs/2207.10666
|
| 35 |
+
* Original GitHub Repository: https://github.com/microsoft/Cream/tree/main/TinyViT
|
| 36 |
+
* **Dataset:** The model was trained on the `iloncka/mosquito-species-classification-dataset`. This is one of a suite of datasets which also includes `iloncka/mosquito-species-detection-dataset` and `iloncka/mosquito-species-segmentation-dataset`. These datasets contain images of various mosquito species, crucial for training accurate identification models. For instance, some datasets include species like *Aedes aegypti*, *Aedes albopictus*, and *Culex quinquefasciatus*, and are annotated for features like normal or smashed conditions.
|
| 37 |
+
* **Pretrain Dataset:** ImageNet-22k, ImageNet-1k
|
| 38 |
|
| 39 |
+
**Model Usage:**
|
| 40 |
|
| 41 |
+
The model can be used for image classification tasks. Below is a code snippet demonstrating how to use the model with the Fastai library:
|
| 42 |
|
| 43 |
+
```python
|
| 44 |
+
from fastai.vision.all import load_learner
|
| 45 |
+
from PIL import Image
|
| 46 |
|
| 47 |
+
# It is assumed that the model has been downloaded locally
|
| 48 |
+
learner = load_learner(model_path)
|
| 49 |
+
_, _, probabilities = learner.predict(image)
|
| 50 |
+
```
|
| 51 |
|
| 52 |
+
**The CulicidaeLab Project:**
|
| 53 |
|
| 54 |
+
The culico-net-cls-v1 model is a component of the larger CulicidaeLab project. This project aims to provide a comprehensive suite of tools for mosquito monitoring and research. Other parts of the project include:
|
|
|
|
| 55 |
|
| 56 |
+
* **Datasets:**
|
| 57 |
+
* `iloncka/mosquito-species-detection-dataset`
|
| 58 |
+
* `iloncka/mosquito-species-segmentation-dataset`
|
| 59 |
+
* `iloncka/mosquito-species-classification-dataset`
|
| 60 |
+
* **Python Library:** https://github.com/iloncka-ds/culicidaelab
|
| 61 |
+
* **Mobile Applications:**
|
| 62 |
+
* - https://gitlab.com/mosquitoscan/mosquitoscan-app
|
| 63 |
+
- https://github.com/iloncka-ds/culicidaelab-mobile
|
| 64 |
+
* **Web Application:** https://github.com/iloncka-ds/culicidaelab-server
|
| 65 |
|
| 66 |
+
**Practical Applications:**
|
| 67 |
+
|
| 68 |
+
The `culico-net-cls-v1` model and the broader `CulicidaeLab` project have several practical applications:
|
| 69 |
+
|
| 70 |
+
* **Integration into Third-Party Products:** The models can be integrated into existing applications for plant and animal identification to expand their functionality to include mosquito recognition.
|
| 71 |
+
* **Embedded Systems (Edge AI):** These models can be optimized for deployment on edge devices such as smart traps, drones, or cameras for in-field monitoring without requiring a constant internet connection.
|
| 72 |
+
* **Accelerating Development:** The pre-trained models can serve as a foundation for transfer learning, enabling researchers to develop systems for identifying other insects or specific mosquito subspecies more efficiently.
|
| 73 |
+
* **Expert Systems:** The model can be used as a "second opinion" tool to assist specialists in quickly verifying species identification.
|
| 74 |
+
|
| 75 |
+
**Acknowledgments:**
|
| 76 |
+
|
| 77 |
+
The development of CulicidaeLab is supported by a grant from the **Foundation for Assistance to Small Innovative Enterprises ([FASIE](https://fasie.ru/))**.
|