DOI: 10.5281/zenodo.4680119
title: Sticky Pi -- Machine Learning Data, Configuration and Models

description: <strong>Dataset for the Machine Learning section of the Sticky Pi project (https://doc.sticky-pi.com/)</strong> Contains the dataset for the three algorithms described in the publication: Universal Insect Detector, Siamese Insect Matcher and Insect Tuboid Classifier. <strong>Universal Insect Detector:</strong> `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 as well as the taxonomic labels `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 <strong>Siamese Insect Matcher</strong> `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 `&lt;device&gt;.&lt;datetime_frame_1&gt;.&lt;datetime_frame_2&gt;.svg` `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 `config.yaml` – The configuration file defining the hyperparameters to train the model <strong>Insect Tuboid Classifier:</strong> `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: `&lt;series_id&gt;/&lt;tuboid_id&gt;/`. 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 – &lt;device&gt;.&lt;UTC_datetime&gt; 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 (&gt;1 =&gt; 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