Metadata-Version: 2.4
Name: culicidaelab
Version: 0.2.2
Summary: A Python library for mosquito detection, segmentation, and classification in images
Author-email: Ilona Kovaleva <iloncka.ds@gmail.com>
Project-URL: Homepage, https://github.com/iloncka/culicidaelab
Project-URL: Bug Tracker, https://github.com/iloncka/culicidaelab/issues
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering :: Image Recognition
Requires-Python: >=3.11
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: torch>=2.3.1
Requires-Dist: torchvision>=0.18.1
Requires-Dist: numpy>=1.24.4
Requires-Dist: tqdm>=4.66.1
Requires-Dist: iopath>=0.1.10
Requires-Dist: pillow>=9.4.0
Requires-Dist: fastai<=2.8.0,>=2.7.0
Requires-Dist: timm>=0.9.0
Requires-Dist: ultralytics>=8.3.0
Requires-Dist: pyyaml>=6.0.0
Requires-Dist: huggingface-hub>=0.16.0
Requires-Dist: datasets==4.0.0
Requires-Dist: omegaconf>=2.3.0
Requires-Dist: python-dotenv>=0.21.0
Requires-Dist: pycocotools>=2.0.7
Requires-Dist: dill>=0.3.8
Requires-Dist: graphviz>=0.20.3
Requires-Dist: appdirs>=1.4.4
Requires-Dist: requests>=2.25.1
Requires-Dist: toml>=0.10.2
Requires-Dist: dotenv>=0.9.9
Requires-Dist: pydantic-settings>=2.9.1
Provides-Extra: examples
Requires-Dist: ipykernel==6.16.2; extra == "examples"
Requires-Dist: ipywidgets>=8.1.7; extra == "examples"
Requires-Dist: jupytext>=1.17.2; extra == "examples"
Provides-Extra: docs
Requires-Dist: culicidaelab[examples]; extra == "docs"
Requires-Dist: mkdocs[i18n]>=1.6.1; extra == "docs"
Requires-Dist: mkdocs-material==9.5.46; extra == "docs"
Requires-Dist: mkdocstrings[python]==0.27.0; extra == "docs"
Requires-Dist: mkdocs-custom-fences>=0.1.2; extra == "docs"
Requires-Dist: mkdocs-gallery>=0.10.4; extra == "docs"
Provides-Extra: test
Requires-Dist: black==23.3.0; extra == "test"
Requires-Dist: mypy==1.4.1; extra == "test"
Requires-Dist: pre-commit==2.21.0; extra == "test"
Requires-Dist: pytest==7.4.4; extra == "test"
Requires-Dist: pytest-cov==4.1.0; extra == "test"
Requires-Dist: pytest-mock==3.14.1; extra == "test"
Requires-Dist: ruff==0.8.0; extra == "test"
Requires-Dist: twine==5.1.1; extra == "test"
Requires-Dist: bandit==1.8.5; extra == "test"
Requires-Dist: types-requests==2.32.4.20250611; extra == "test"
Requires-Dist: types-pyyaml==6.0.12.20250516; extra == "test"
Requires-Dist: types-toml==0.10.8.20240310; extra == "test"
Requires-Dist: gputil==1.4.0; extra == "test"
Requires-Dist: requests-mock>=1.12.1; extra == "test"
Requires-Dist: tabulate>=0.9.0; extra == "test"
Provides-Extra: dev
Requires-Dist: culicidaelab[docs]; extra == "dev"
Requires-Dist: culicidaelab[test]; extra == "dev"
Dynamic: license-file

<div align="center">

# CulicidaeLab 🦟

**A configuration-driven Python library for advanced mosquito analysis, featuring pre-trained models for detection, segmentation, and species classification.**

</div>

<p align="center">
  <a href="https://pypi.org/project/culicidaelab/"><img alt="PyPI" src="https://img.shields.io/pypi/v/culicidaelab?color=blue"></a>

<a href="#">
  <a href="https://github.com/astral-sh/ruff"><img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json" alt="Ruff" /></a>
</p>

---

`culicidaeLab` python library provides a robust, extensible framework designed to streamline the pipeline of mosquito image analysis. Built on a powerful configuration system, it allows researchers and developers to easily manage datasets, experiment with models, and process images for classification, detection, and segmentation tasks. `culicidaelab` library is part of the is a part of the CulicidaeLab Ecosystem.

## CulicidaeLab Ecosystem Architecture

```mermaid
flowchart TD

subgraph L0 [" "]

    %% Define layers with subgraphs

    subgraph L1 ["Data Layer"]

        DS1["🦟 mosquito_dataset_46_3139<br/>Base Diversity Dataset<br/>(46 species, 3139 unique images)<br/>📄 CC-BY-SA-4.0"]

        DS2["📊 mosquito-species-<br/>classification-dataset<br/>📄 CC-BY-SA-4.0"]

        DS3["🔍 mosquito-species-<br/>detection-dataset<br/>📄 CC-BY-SA-4.0"]

        DS4["✂️ mosquito-species-<br/>segmentation-dataset<br/>📄 CC-BY-SA-4.0"]

    end



    subgraph L2 ["AI Model Layer"]

        subgraph M_COLLECTION ["Top-5 Model Collection"]

            M4["📊 exp_7_new_bg_simple-subs_1_v_5<br/>pvt_v2_b0.in1k_ep_60<br/>(Classification)<br/>📄 Apache 2.0"]

        end
		subgraph M_DEFAULT ["Top-1 Models used as default in 'culicidaelab'"]
        M1["📊 culico-net-cls-v1<br/>(Classification)<br/>📄 Apache 2.0"]

        M2["🔍 culico-net-det-v1<br/>(Detection)<br/>📄 AGPL-3.0"]

        M3["✂️ culico-net-segm-v1-nano<br/>(Segmentation)<br/>📄 Apache 2.0"]
		end
    end



    subgraph L3 ["Application Layer"]

        APP1["🐍 culicidaelab<br/>Python Library<br/>(Core ML functionality) <br/>📄 AGPL-3.0"]

        APP2["🌐 culicidaelab-server<br/>Web Application<br/>(API services)<br/>📄 AGPL-3.0"]

        APP3["📸 culicidaelab-mobile<br/>Mobile Application<br/><br/>📄 AGPL-3.0"]

    end



    subgraph L4 ["API Service Layer"]

        S1["🗲 Prediction Service<br/>(ML inference)"]

        S2["💾 Observation Service<br/>(Data storage & retrieval)"]

        S3["🗺️ Map Service<br/>(Geospatial visualization)"]

        S4["🦟 Mosquito Gallery Service<br/>"]
       
        S5["💊 Diseases Gallery Service<br/>"]

    end

  end

   

    %% Dataset derivation and training flows

    DS1 -.->|"derives"| DS2

    DS1 -.->|"derives"| DS3

    DS1 -.->|"derives"| DS4

    DS2 -->|"used for train"| M1

    DS3 -->|"used for train"| M2

    DS4 -->|"used for train"| M3

    DS2 -->|"used for train"| M4



    %% Model integration

    M1 -->|"integrated into"| APP1

    M2 -->|"integrated into"| APP1

    M3 -->|"integrated into"| APP1

    M4 -->|"integrated into"| APP3



    %% Data source for gallery

    DS1 -->|"provides photos"| APP2
	  DS1 -->|"provides photos"| APP3


    %% Library to server integration

    APP1 -->|"powers"| APP2


    %% Service provisioning

    APP2 -->|"hosts"| S1

    APP2 -->|"hosts"| S2

    APP2 -->|"hosts"| S3

    APP2 -->|"hosts"| S4

	  APP2 -->|"hosts"| S5

    %% Mobile app service consumption

    APP3 <-->|"API calls"| S1

    APP3 <-->|"API calls"| S2

    APP3 -->|"WebView"| S3



    %% Styling

    classDef dataLayer fill:#e3f2fd,stroke:#1976d2,stroke-width:2px

    classDef modelLayer fill:#f3e5f5,stroke:#7b1fa2,stroke-width:2px

    classDef appLayer fill:#e8f5e8,stroke:#388e3c,stroke-width:2px

    classDef serviceLayer fill:#fff3e0,stroke:#f57c00,stroke-width:2px
   
    classDef collections fill:#f4dbf8,stroke:#9b3ac5,stroke-width:1px,stroke-dasharray:5,5	 



    classDef dataset fill:#bbdefb,stroke:#1565c0,stroke-width:2px

    classDef model fill:#e1bee7,stroke:#8e24aa,stroke-width:2px

    classDef application fill:#c8e6c9,stroke:#43a047,stroke-width:2px

    classDef service fill:#ffe0b2,stroke:#fb8c00,stroke-width:2px   



    class L1 dataLayer

    class L2 modelLayer

    class L3 appLayer

    class L4 serviceLayer
   
	  class L5 collections
   
    class DS1,DS2,DS3,DS4 dataset

    class M1,M2,M3,M4 model

    class APP1,APP2,APP3 application

    class S1,S2,S3,S4,S5 service

    class M_DEFAULT,M_COLLECTION collections
   
```

An open-source system for mosquito research and analysis includes components:

- **Data**:

  - Base [diversity dataset (46 species, 3139 images](https://huggingface.co/datasets/iloncka/mosquito_dataset_46_3139)) under CC-BY-SA-4.0 license.
  - Specialized derivatives: [classification](https://huggingface.co/datasets/iloncka/mosquito-species-classification-dataset), [detection](https://huggingface.co/datasets/iloncka/mosquito-species-detection-dataset), and [segmentation](https://huggingface.co/datasets/iloncka/mosquito-species-segmentation-dataset) datasets under CC-BY-SA-4.0 licenses.

- **Models**:

  - Top-1 models (see reports), used as default by `culicidaelab` library: [classification (Apache 2.0)](https://huggingface.co/iloncka/culico-net-cls-v1), [detection (AGPL-3.0)](https://huggingface.co/iloncka/culico-net-det-v1), [segmentation (Apache 2.0)](https://huggingface.co/iloncka/culico-net-segm-v1-nano)
  - [Top-5 classification models collection](https://huggingface.co/collections/iloncka/mosquito-classification-17-top-5-68945bf60bca2c482395efa8) with accuracy >90% for 17 mosquito species.

- **Protocols**: All training parameters and metrics available at:

  - [Detection model reports](https://gitlab.com/mosquitoscan/experiments-reports-detection-models)
  - [Segmentation model reports](https://gitlab.com/mosquitoscan/experiments-reports-segmentation-models)
  - [Classification experiment reports - 1st round](https://gitlab.com/iloncka/mosal-reports)
  - [Classification experiment reports -2nd round](https://gitlab.com/mosquitoscan/experiments-reports)

- **Applications**:

  - [Python library (AGPL-3.0)](https://github.com/iloncka-ds/culicidaelab) providing core ML functionality
  - [Web server (AGPL-3.0)](https://github.com/iloncka-ds/culicidaelab-server) hosting API services
  - [Mobile app (AGPL-3.0)](https://gitlab.com/mosquitoscan/mosquitoscan-app) for field use with optimized models

These components form a cohesive ecosystem where datasets used for training models that power applications, the Python library provides core functionality to the web server, and the server exposes services consumed by the mobile application. All components are openly licensed, promoting transparency and collaboration.

This integrated approach enables comprehensive mosquito research, from data collection to analysis and visualization, supporting both scientific research and public health initiatives.

## Key Features of the `culicidaelab` library

- **Configuration-Driven Workflow**: Manage all settings—from file paths to model parameters—through simple YAML files. Override defaults easily for custom experiments.
- **Ready-to-Use Models**: Leverage pre-trained models for:
  - **Species Classification**: Identify mosquito species using a high-accuracy classifier.
  - **Mosquito Detection**: Localize mosquitoes in images with a YOLO-based detector.
  - **Instance Segmentation**: Generate precise pixel-level masks with a SAM-based segmenter.
- **Unified API**: All predictors share a consistent interface (`.predict()`, `.visualize()`, `.evaluate()`) for a predictable user experience.
- **Automatic Resource Management**: The library intelligently manages local storage, automatically downloading and caching model weights and datasets on first use.
- **Extensible Provider System**: Seamlessly connect to data sources. A `HuggingFaceProvider` is built-in, with an easy-to-implement interface for adding more providers.
- **Powerful Visualization**: Instantly visualize model outputs with built-in, configurable methods for drawing bounding boxes, classification labels, and segmentation masks.

## Requirements

### Hardware Requirements

**Processor (CPU):** Any modern x86-64 CPU.

**Memory (RAM):** Minimum 2 GB. 8 GB or more is recommended for processing large datasets or using more complex models.

**Graphics Card (GPU):** An NVIDIA GPU with CUDA support is highly recommended for a significant performance increase in deep learning model operations, especially for detection and segmentation but not essential for classification (see [performance logs](https://github.com/iloncka-ds/culicidaelab/tree/main/tests/performance/performance_logs) ang [notebook](https://colab.research.google.com/drive/1JdfxSQmtrJND4mNUctOkY7Kt0yvbO0eV?usp=sharing)). For the SAM model, a GPU is virtually essential for acceptable performance. Minimum video memory is 2 GB; 4 GB or more is recommended.

**Hard Drive:** At least 10 GB of free space to install the library, dependencies, download pre-trained models, and store processed data.

### Software Requirements:
  Operating Systems (tested):
  - Windows 10/11
  - Linux 22.04+
  Software:
  - for Linux needed libgl1 package to be installed
  - Git
  - Python 3.11
  - uv 0.8.13
  Python packages:
  - PyTorch 2.3.1+
  - FastAI 2.7.0 - 2.8.0
  - Ultralytics 8.3.0+
  - HuggingFace Hub 0.16.0+
  - Datasets 4.0.0
  - Pillow 9.4.0
  - Pydantic 2.0.0+
    Full list of requirements: [requirements.txt](https://github.com/iloncka-ds/culicidaelab/blob/main/requirements.txt)
    Development requirements: [requirements-dev.txt](https://github.com/iloncka-ds/culicidaelab/blob/main/dev-requirements.txt)


## Installation
For general usage with Python scripts or in Google Colab, you can install `culicidaelab` using pip:

```bash
pip install culicidaelab
```

If needed run examples in the Jupyter notebooks in local environment:

```bash
pip install culicidaelab[examples]
```

If needed build documentation in local environment:

```bash
pip install culicidaelab[docs]
```

If needed run tests in local environment:

```bash
pip install culicidaelab[test]
```

To get a **development environment** running:

  1. Clone the repository:

     ```bash
     git clone https://github.com/iloncka-ds/culicidaelab.git
     cd culicidaelab
     ```
  2. Install dependencies with `uv` (recommended):

     ```bash
     uv venv -p 3.11
     uv sync -p 3.11
     uv cache clean
     # This installs the library in editable mode and includes all dev tools
     uv pip install -e .[dev]
     ```
     Or with `pip`:

     ```bash
     python -m venv .venv
     pip install --upgrade pip
     pip install -e .[dev]
     pip cache purge
     ```
  3. Set up pre-commit hooks:

     ```bash
     pre-commit install
     ```

     This will run linters and formatters automatically on each commit to ensure code quality and consistency.

## Quick Start

Here's how to classify the species of a mosquito in just a few lines of code. The library will automatically download the necessary model on the first run.

```python

from culicidaelab import MosquitoClassifier, get_settings

# 1. Get the central settings object
# This loads all default configurations for the library.
settings = get_settings()

# 2. Instantiate the classifier
# The settings object knows how to configure the classifier.
classifier = MosquitoClassifier(settings, load_model=True)

# 3. Make a prediction
# The model is lazy-loaded (downloaded and loaded into memory) here.
predictions = classifier.predict("path/to/your/image.jpg")

# 5. Print the results
# The output is a list of (species_name, confidence_score) tuples.
print("Top 3 Predictions:")
for species, confidence in predictions[:3]:
    print(f"- {species}: {confidence:.4f}")

# Example Output:
# Top 3 Predictions:
# - Aedes aegypti: 0.9876
# - Aedes albopictus: 0.0112
# - Culex quinquefasciatus: 0.0009
```

## Practical Applications

`CulicidaeLab` is more than just a set of models; it's a powerful engine for building real-world solutions. Here are some of the ways it can be applied:

- **Automation in Scientific Laboratories:**

  - **Bulk Data Processing:** Automatically analyze thousands of images from camera traps or microscopes to assess mosquito populations without manual intervention.
  - **Reproducible Research:** Standardize the data analysis process, allowing other scientists to easily reproduce and verify research results published using the library.
- **Integration into Governmental and Commercial Systems:**

  - **Epidemiological Surveillance:** Use the library as the core "engine" for national or regional monitoring systems to track vector-borne disease risks.
  - **Custom Solution Development:** Rapidly prototype and create specialized software products for pest control services, agro-industrial companies, or environmental organizations.
- **Advanced Analytics and Data Science:**

  - **Geospatial Analysis:** Write scripts to build disease vector distribution maps by processing geotagged images.
  - **Predictive Modeling:** Use the library's outputs as features for larger models that forecast disease outbreaks based on vector presence and density.

## Documentation

For complete guides, tutorials, and the full API reference, **[visit the documentation site](https://iloncka-ds.github.io/culicidaelab/)**.

The documentation includes:

- In-depth installation and configuration guides.
- Detailed tutorials for each predictor.
- Architectural deep-dives for contributors.
- A full, auto-generated API reference.

## Contributing

Contributions are what make the open-source community such an amazing place to learn, inspire, and create. Any contributions you make are **greatly appreciated**.

Please see our **[Contributing Guide](https://github.com/iloncka-ds/culicidaelab/blob/main/CONTRIBUTING.md)** for details on our code of conduct, development setup, and the pull request process.

## Acknowledgments

CulicidaeLab development is  supported by a grant from the [**Foundation for Assistance to Small Innovative Enterprises (FASIE)**](https://fasie.ru/)

## License

This project is distributed under the  AGPL-3.0 License. See the [LICENSE](https://github.com/iloncka-ds/culicidaelab/blob/main/LICENSE) file for more information.

## Citation

If you use `CulicidaeLab` in your research, please cite it as follows:

```bibtex
@software{culicidaelab2024,
  author = {Ilona Kovaleva},
  title = {{CulicidaeLab: A Configuration-Driven Python Library for Mosquito Analysis}},
  year = {2024},
  publisher = {GitHub},
  journal = {GitHub repository},
  url = {https://github.com/iloncka-ds/culicidaelab}
}
```

## Contact

- **Issues**: Please use the [GitHub issue tracker](https://github.com/iloncka-ds/culicidaelab/issues).
- **Email**: [iloncka.ds@gmail.com](mailto:iloncka.ds@gmail.com)
