Python Library for Nodes which connect to the Zauberzeug Learning Loop
Project description
Learning Loop Node
This Python library helps to write Nodes that interact with the Zauberzeug Learning Loop. There are 4 types of Nodes:
Type | Purpose |
---|---|
Trainer | Runs training using the latest training data |
Detector | Loads latest models from Loop and performs inference |
Annotator | Used for custom annotation inside the Loop |
Converter | Converts between different model formats |
General Usage
To start a node you have to implement the logic by inheriting from the corresponding base logic class. We provide samples in the 'mock' folders and recommend to follow that scheme. A complete trainer and detector example can be found here.
Environment variables
You can configure connection to our Learning Loop by specifying the following environment variables before starting:
Name | Alias | Purpose | Required by |
---|---|---|---|
LOOP_HOST | HOST | Learning Loop adress (e.g. learning-loop.ai) | all |
LOOP_USERNAME | USERNAME | Learning Loop user name | all |
LOOP_PASSWORD | PASSWORD | Learning Loop password | all |
LOOP_ORGANIZATION | ORGANIZATION | Organization name | Detector |
LOOP_PROJECT | PROJECT | Project name | Detector |
Testing
We use github actions for CI. Tests can also be executed locally by running
LOOP_HOST=XXXXXXXX LOOP_USERNAME=XXXXXXXX LOOP_PASSWORD=XXXXXXXX python -m pytest -v
from learning_loop_node/learning_loop_node
Detector Node
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. They provide REST and Socket.io APIs to run inference on images. The processed images can automatically be used for active learning: e.g. uncertain predictions will be send to the Learning Loop.
Trainer Node
Trainers fetch the images and anntoations from the Learning Loop to train new models.
- if the command line tool "jpeginfo" is installed, the downloader will drop corrupted images automatically
Converter Node
A Conveter Node converts models from one format into another.
Annotator Node
...
Test operability
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",
...
}
]
}
About Models (the currency between Nodes)
- 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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for learning_loop_node-0.8.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c30d2209d3bcafb335ebb63a7200fafda80053ff55e023f16aa0ea9ecf279290 |
|
MD5 | a801779a39219df7f9daee9b49786147 |
|
BLAKE2b-256 | 5a3bab35363fcb5927b33f0c7ba4c150acfa54d78da43290a6a74e656db7ba6a |