Unsuccessful remap_labels

[sizov-kirill]

Expected Behaviour

Successful remapping

Current Behaviour

Transformation fails:

2021-06-27 19:53:21,789 INFO: Loading the project...
2021-06-27 19:54:15,506 INFO: Transforming the project...
2021-06-27 19:54:15,506 ERROR: 22
Traceback (most recent call last):
  File "<path/to/datumaro>/venv/bin/datum", line 33, in <module>
    sys.exit(load_entry_point('datumaro', 'console_scripts', 'datum')())
  File "<path/to/datumaro>/datumaro/cli/__main__.py", line 126, in main
    return args.command(args)
  File "<path/to/datumaro>/datumaro/cli/contexts/project/__init__.py", line 685, in transform_command
    dataset.transform_project(
  File "<path/to/datumaro>/datumaro/components/project.py", line 294, in transform_project
    transformed = self.transform(method, **method_kwargs)
  File "<path/to/datumaro>/datumaro/components/project.py", line 256, in transform
    return method(self, *args, **kwargs)
  File "<path/to/datumaro>/datumaro/plugins/transforms.py", line 502, in __init__
    dst_mask_cat.colormap = {
  File "<path/to/datumaro>/datumaro/plugins/transforms.py", line 503, in <dictcomp>
    id: src_mask_cat.colormap[id]
KeyError: 22

Steps to Reproduce

1. Create a dataset containing labels categories and masks categories of different lengths (e.g Pascal VOC 2012)
2. Try to perform remap labels:

datum create
datum add path -f voc <path/to/dataset>
datum transform -t remap_labels -- -l cat:label_0

Environment

• Git hash commit (git log -1): 7e35c8