Is there a way of passing a group_id back into CVAT after automatic annotation?

[MPiotr]

Hi,
I am trying to run OpenPose as an automatic annotation tool. Here is an example of a JSON returned by model

[{'points': [359, 186], 'type': 'points', 'label': 'Nose', 'group_id': 0, 'confidence': '1'}, 
{'points': [345, 213], 'type': 'points', 'label': 'Neck', 'group_id': 0, 'confidence': '1'}, 
...
{'points': [300, 200], 'type': 'points', 'label': 'Neck', 'group_id': 1, 'confidence': '1'}, 
...] 

Current Behaviour

The points are created in cvat_ui, however the group_id affects nothing: all the points has the same group

Expected Behaviour

I would expect that point groups in ui are correspond to group_id filed in JSON.

Your Environment

• Git hash commit: 31f6234
• Docker 20.10.8
• Are you using Docker Swarm or Kubernetes? No
• Ubuntu 18.04

[bsekachev]

Hi,

In the file cvat/apps/lambda_manager/views.py LamdaJob::_call_detector this code is responsible for preparing results:

results.append_shape({
                        "frame": frame,
                        "label_id": label_id,
                        "type": anno["type"],
                        "occluded": False,
                        "points": anno["points"],
                        "z_order": 0,
                        "group": None,
                        "attributes": [],
                        "source": "auto"
                    })

As you can see it sets group as None always. That is because current detectors does not support groups.
Maybe better option is "group": anno["group_id"] if "group_id" in anno else None

[nmanovic]

@MPiotr , if you can help and prepare a PR, it will be awesome

[zwfcrazy]

Hi, just a reminder that this hasn't been fixed yet.
Though I can fix this in my own deployment, I strongly recommend you fix this asap as group_id is useful when we have multiple instances of person keypoints.

[nmanovic]

@zwfcrazy , could you please send us a PR? We really like when community contributes into the project and help us to improve it.

[zwfcrazy]

@nmanovic It seems that making the fix mentioned by @bsekachev is not enough, I am trying to figure out the cause.

[zwfcrazy]

@nmanovic It seems that making the fix mentioned by @bsekachev is not enough, I am trying to figure out the cause.

Found the cause. When implementing an auto-annotation function, be aware group_id should start from 1 instead of 0. This is because group_id=0 is equivalent to no group.