Image Classification
fastai
ONNX
timm
Transformers
iloncka commited on
Commit
c876804
·
verified ·
1 Parent(s): 8de6623

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +64 -20
README.md CHANGED
@@ -1,33 +1,77 @@
1
- ---
2
- license: apache-2.0
3
- tags:
4
- - fastai
5
- ---
 
 
 
 
 
 
 
 
 
 
 
 
6
 
7
- # Amazing!
8
 
9
- 🥳 Congratulations on hosting your fastai model on the Hugging Face Hub!
10
 
11
- # Some next steps
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
- 2. Create a demo in Gradio or Streamlit using 🤗 Spaces ([documentation here](https://huggingface.co/docs/hub/spaces)).
15
 
16
- 3. Join the fastai community on the [Fastai Discord](https://discord.com/invite/YKrxeNn)!
 
 
 
 
 
 
 
 
 
 
17
 
18
- Greetings fellow fastlearner 🤝! Don't forget to delete this content from your model card.
19
 
 
20
 
21
- ---
 
 
22
 
 
 
 
 
23
 
24
- # Model card
25
 
26
- ## Model description
27
- More information needed
28
 
29
- ## Intended uses & limitations
30
- More information needed
 
 
 
 
 
 
 
31
 
32
- ## Training and evaluation data
33
- More information needed
 
 
 
 
 
 
 
 
 
 
 
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/))**.