Skip to content

feat: add the model_labels in the list models endpoint #70

@dido18

Description

@dido18

Description

The get models does not return the models_labels in the response:

GET curl 127.0.0.1:8800/v1/models

"models":[
      {
         "id":"face-detection",
         "name":"Lightweight-Face-Detection",
         "description":"Face bounding box detection. This model is trained on the WIDER FACE dataset and can detect faces in images.",
         "runner":"brick",
         "brick_ids":[
            "arduino:object_detection",
            "arduino:video_object_detection"
         ],
         "metadata":{
            "ei-gpu-mode":"false",
            "source":"qualcomm-ai-hub",
            "source-model-id":"face-det-lite",
            "source-model-url":"https://aihub.qualcomm.com/models/face_det_lite"
         },
         "model_configuration":{
            "EI_OBJ_DETECTION_MODEL":"/models/ootb/ei/lw-face-det.eim"
         }
      },
...}

expected

"models":[
      {
         "id":"face-detection",
         "name":"Lightweight-Face-Detection",
         "description":"Face bounding box detection. This model is trained on the WIDER FACE dataset and can detect faces in images.",
         "runner":"brick",
         "model_labels": [
               "face"
          ],
         "brick_ids":[
            "arduino:object_detection",
            "arduino:video_object_detection"
         ],
         "metadata":{
            "ei-gpu-mode":"false",
            "source":"qualcomm-ai-hub",
            "source-model-id":"face-det-lite",
            "source-model-url":"https://aihub.qualcomm.com/models/face_det_lite"
         },
         "model_configuration":{
            "EI_OBJ_DETECTION_MODEL":"/models/ootb/ei/lw-face-det.eim"
         }
      },
...}

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions