Skip to content

atubito98/Mosaico_train

Repository files navigation

Mosaico_train

This repository contains the training and evaluation code for the mosquito species classification model powering the Mosaico Webapp. The core is a convolutional neural network (CNN) with ~50 million parameters, fine-tuned on a private dataset collected by the Istituto Superiore di Sanità (ISS).

📊 Dataset

  • Private dataset of mosquito images from across Italy and some international samples.
  • Covers 15 distinct mosquito species.
  • Data collected and annotated by expert entomologists.

🧠 Model & Approach

  • Base CNN model fine-tuned using Modified Evidential Deep Learning (EDL).
  • Designed for Open Set Classification to handle unknown or novel classes.
  • Provides calibrated uncertainty estimates on predictions, important for expert review and feedback loops.

🚀 Features

  • Training scripts with configurable hyperparameters.
  • Evaluation and test pipelines producing metrics and uncertainty calibration analysis.
  • Integration-ready checkpoints for deployment in the main Mosaico web platform.

About

Code for training Mosaico mosquito classifier: fine-tunes a 50M-parameter CNN on a 15-species ISS dataset with modified Evidential Deep Learning for open-set recognition and uncertainty estimation.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors