This Python library helps you to write your own Detection Nodes, Training Nodes and Converter Nodes for the Zauberzeug Learning Loop.
You can configure connection to our Learning Loop by specifying the following environment variables before starting:
- LOOP_HOST=learning-loop.ai
- LOOP_USERNAME=
- LOOP_PASSWORD=
Detector Nodes are normally deployed on edge devices like robots or machinery but can also run in the cloud to provide backend services for an app or similar. These nodes register themself at the Learning Loop to make model deployments very easy. They also provide REST and Socket.io APIs to run inferences. By default the images will automatically used for active learning: high uncertain predictions will be submitted to the Learning Loop inbox.
- LOOP_ORGANIZATION=
- LOOP_PROJECT=
Trainers fetch the images and anntoations from the Learning Loop to generate new and improved models.
- if the command line tool "jpeginfo" is installed, the downloader will drop corrupted images automatically
A Conveter Node converts models from one format into another.
Assumend there is a Converter Node which converts models of format 'format_a' into 'format_b'.
Upload a model with
curl --request POST -F 'files=@my_model.zip' https://learning-loop.ai/api/zauberzeug/projects/demo/format_a
The model should now be available for the format 'format_a'
curl "https://learning-loop.ai/api/zauberzeug/projects/demo/models?format=format_a"
{
"models": [
{
"id": "3c20d807-f71c-40dc-a996-8a8968aa5431",
"version": "4.0",
"formats": [
"format_a"
],
"created": "2021-06-01T06:28:21.289092",
"comment": "uploaded at 2021-06-01 06:28:21.288442",
...
}
]
}
but not in the format_b
curl "https://learning-loop.ai/api/zauberzeug/projects/demo/models?format=format_b"
{
"models": []
}
Connect the Node to the Learning Loop by simply starting the container.
After a short time the converted model should be available as well.
curl https://learning-loop.ai/api/zauberzeug/projects/demo/models?format=format_b
{
"models": [
{
"id": "3c20d807-f71c-40dc-a996-8a8968aa5431",
"version": "4.0",
"formats": [
"format_a",
"format_b",
],
"created": "2021-06-01T06:28:21.289092",
"comment": "uploaded at 2021-06-01 06:28:21.288442",
...
}
]
}
- Models are packed in zips and saved on the Learning Loop (one for each format)
- Nodes and users can upload and download models with which they want to work
- In each zip there is a file called
model.json
which contains the metadata to interpret the other files in the package - for base models (pretrained models from external sources) no
model.json
has to be sent, ie. these models should simply be zipped in such a way that the respective trainer can work with them. - the loop adds or corrects the following properties in the
model.json
after receiving; it also creates the file if it is missing:host
: uri to the looporganization
: the ID of the organizationproject
: the id of the projectversion
: the version number that the loop assigned for this model (e.g. 1.3)id
: the model UUID (currently not needed by anyone, since host, org, project, version clearly identify the model)format
: the format e.g. yolo, tkdnn, yolor etc.
- Nodes add properties to
model.json
, which contains all the information which are needed by subsequent nodes. These are typically the properties:resolution
: resolution in which the model expects images (asint
, since the resolution is mostly square - later,resolution_x`` resolution_y
would also be conceivable orresolutions
to give a list of possible resolutions)categories
: list of categories with name, id, (later also type), in the order in which they are used by the model -- this is neccessary to be robust about renamings