Fix label comparison in voc format (#1382)

main
zhiltsov-max 6 years ago committed by GitHub
parent 13b8cfa46e
commit 819b81d7f3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -35,6 +35,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
- Merge is allowed for points, but clicks on points conflict with frame dragging logic
- Removed objects are visible for search
- Add missed task_id and job_id fields into exception logs for the new UI (https://github.com/opencv/cvat/pull/1372)
- VOC format exports Upper case labels correctly in lower case (https://github.com/opencv/cvat/pull/1379)
- Fixed polygon exporting bug in COCO dataset (https://github.com/opencv/cvat/issues/1387)
- Task creation from remote files (https://github.com/opencv/cvat/pull/1392)

@ -1,5 +1,5 @@
# Copyright (C) 2019 Intel Corporation
# Copyright (C) 2020 Intel Corporation
#
# SPDX-License-Identifier: MIT
@ -110,8 +110,8 @@ class _Converter:
self._images_dir = images_dir
def get_label(self, label_id):
return self._extractor.categories()[AnnotationType.label] \
.items[label_id].name
return self._strip_label(self._extractor. \
categories()[AnnotationType.label].items[label_id].name)
def save_subsets(self):
subsets = self._extractor.subsets()
@ -426,7 +426,7 @@ class _Converter:
label_map = OrderedDict()
label_map['background'] = [None, [], []]
for item in labels.items:
label_map[item.name] = [None, [], []]
label_map[self._strip_label(item.name)] = [None, [], []]
elif label_map_source in [LabelmapType.guess.name, None]:
# generate colormap for union of VOC and input dataset labels
@ -489,7 +489,7 @@ class _Converter:
def _make_label_id_map(self):
source_labels = {
id: label.name for id, label in
id: self._strip_label(label.name) for id, label in
enumerate(self._extractor.categories().get(
AnnotationType.label, LabelCategories()).items)
}

@ -649,7 +649,7 @@ class VocConverterTest(TestCase):
def categories(self):
label_cat = LabelCategories()
label_cat.add('label_1')
label_cat.add('Label_1') # should become lowercase
label_cat.add('label_2')
return {
AnnotationType.label: label_cat,

Loading…
Cancel
Save