def convert(self, input_data, output_data, format, is_dir=True, **kwargs):
if isinstance(format, str):
format = Format.from_string(format)
if format == Format.JSON:
self.convert_to_json(input_data, output_data)
elif format == Format.JSON_MIN:
self.convert_to_json_min(input_data, output_data, is_dir=is_dir)
elif format == Format.CSV:
header = kwargs.get('csv_header', True)
sep = kwargs.get('csv_separator', ',')
self.convert_to_csv(input_data, output_data, sep=sep, header=header, is_dir=is_dir)
elif format == Format.TSV:
header = kwargs.get('csv_header', True)
sep = kwargs.get('csv_separator', '\t')
self.convert_to_csv(input_data, output_data, sep=sep, header=header, is_dir=is_dir)
elif format == Format.CONLL2003:
self.convert_to_conll2003(input_data, output_data, is_dir=is_dir)
elif format == Format.COCO:
image_dir = kwargs.get('image_dir')
self.convert_to_coco(input_data, output_data, output_image_dir=image_dir, is_dir=is_dir)
elif format == Format.VOC:
image_dir = kwargs.get('image_dir')
self.convert_to_voc(input_data, output_data, output_image_dir=image_dir, is_dir=is_dir)
elif format == Format.BRUSH_TO_NUMPY:
image_dir = kwargs.get('image_dir')
brush.convert_task_dir(input_data, output_data, out_format='numpy')
elif format == Format.BRUSH_TO_PNG:
image_dir = kwargs.get('image_dir')
brush.convert_task_dir(input_data, output_data, out_format='png')
def convert(self, input_data, output_data, format, is_dir=True, **kwargs):
if isinstance(format, str):
format = Format.from_string(format)
if format == Format.JSON:
self.convert_to_json(input_data, output_data, is_dir=is_dir)
elif format == Format.JSON_MIN:
self.convert_to_json_min(input_data, output_data, is_dir=is_dir)
elif format == Format.CSV:
header = kwargs.get('csv_header', True)
sep = kwargs.get('csv_separator', ',')
self.convert_to_csv(input_data,
output_data,
sep=sep,
header=header,
is_dir=is_dir)
elif format == Format.TSV:
header = kwargs.get('csv_header', True)
sep = kwargs.get('csv_separator', '\t')
self.convert_to_csv(input_data,
output_data,
sep=sep,
header=header,
is_dir=is_dir)
elif format == Format.CONLL2003:
self.convert_to_conll2003(input_data, output_data, is_dir=is_dir)
elif format == Format.COCO:
image_dir = kwargs.get('image_dir')
self.convert_to_coco(input_data,
output_data,
output_image_dir=image_dir,
is_dir=is_dir)
elif format == Format.VOC:
image_dir = kwargs.get('image_dir')
self.convert_to_voc(input_data,
output_data,
output_image_dir=image_dir,
is_dir=is_dir)
elif format == Format.BRUSH_TO_NUMPY:
items = self.iter_from_dir(
input_data) if is_dir else self.iter_from_json_file(input_data)
brush.convert_task_dir(items, output_data, out_format='numpy')
elif format == Format.BRUSH_TO_PNG:
items = self.iter_from_dir(
input_data) if is_dir else self.iter_from_json_file(input_data)
brush.convert_task_dir(items, output_data, out_format='png')
elif format == Format.ASR_MANIFEST:
items = self.iter_from_dir(
input_data) if is_dir else self.iter_from_json_file(input_data)
convert_to_asr_json_manifest(items,
output_data,
data_key=self._data_keys[0],
project_dir=self.project_dir,
upload_dir=self.upload_dir)
