Skip to content
This repository has been archived by the owner on Aug 21, 2024. It is now read-only.

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
cdpath committed Dec 20, 2022
1 parent 1a3abe3 commit e8195c7
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion label_studio_converter/converter.py
Original file line number Diff line number Diff line change
Expand Up @@ -267,6 +267,9 @@ def _get_supported_formats(self):
all_formats.remove(Format.BRUSH_TO_PNG.name)
if not (('Audio' in input_tag_types or 'AudioPlus' in input_tag_types) and 'TextArea' in output_tag_types):
all_formats.remove(Format.ASR_MANIFEST.name)
if ('Image' in input_tag_types and ('BrushLabels' in output_tag_types or 'brushlabels' in output_tag_types or
'Brush' in output_tag_types and 'Labels' in output_tag_types)):
all_formats.ad(Format.COCO.name)

return all_formats

Expand Down Expand Up @@ -554,7 +557,7 @@ def add_image(images, width, height, image_id, image_path):
for label in labels:

category_name = None
for key in ['rectanglelabels', 'polygonlabels', 'brushlabels' 'labels']:
for key in ['rectanglelabels', 'polygonlabels', 'brushlabels', 'labels']:
if key in label and len(label[key]) > 0:
category_name = label[key][0]
break
Expand Down

0 comments on commit e8195c7

Please sign in to comment.