Sticky Pi -- Machine Learning Data, Configuration and Models

Dataset for the Machine Learning section of the Sticky Pi project (https://doc.sticky-pi.com/)

Contains the dataset for the three algorithms described in the publication: Universal Insect Detector, Siamese Insect Matcher and Insect Tuboid Classifier.

Universal Insect Detector:

`universal_insect_detector/` contains training/validation data, configuration files to train the model, and the model as trained and used for publication.

`data/` – A set of svg images that contain the embedded jpg raw image, and a set of non-intersecting polygon around the labelled insects

`output/`

`model_final.pth` – the model as trained for the publication

`config/`

`config.yaml `– The configuration file defining the hyperparameters to train the model

`mask_rcnn_R_101_C4_3x.yaml` – the base configuration file from which config is derived

Siamese Insect Matcher

`siamese_insect_matcher/` contains training/validation data, configuration files to train the model, and the model as trained and used for publication.

`data/` – a set of svg images that contain two embedded jpg raw images vertically stacked corresponding to two frames in a series. Each predicted insect is labelled as a polygon. Insects that are labelled as the same instance, between the two frames, are grouped (i.e. SVG group). The filename of each image is `...svg`

`output/`

`model_final.pth` – the model as trained for the publication

`config/`

`config.yaml` – The configuration file defining the hyperparameters to train

Insect Tuboid Classifier:

`insect_tuboid_classifier/` contains images of insect tuboid, a database file describing their taxonomy, a configuration file to train the model, and the model as trained and used for publication.

`data/`

`database.db`: a sqlite file with a single table `ANNOTATIONS`. The table maps a unique identifier of each tuboid (tuboid_id) to a set of manually annotated taxonomic variables.

A directory tree of the form: `//`. Each terminal directory contains:

`tuboid.jpg` – a jpeg image made of 224 x 224 tiles representing all the shots in a tuboid, left to right, top to bottom – might be padded with empty images

`metadata.txt` – a csv text file with columns:

parrent_image_id – .

X – the X coordinates of the object centroid

Y – the Y coordinates of the object centroid

scale – The scaling factor applied between the original and image and the 224 x 224 tile (>1 => image was enlarged)

`context.jpg` – a representation of the first whole image of a series, with a box around the first tuboid shot (this is for debugging/labelling purposes)

`output/`

`model_final.pth` – the model as trained for the publication

config/

`config.yaml` – The configuration file defining the hyperparameters to train the model as well as the taxonomic labels