Coretex
coretex.entities.conversion.base_converter.BaseConverter Class Reference
Inheritance diagram for coretex.entities.conversion.base_converter.BaseConverter:
coretex.entities.conversion.converters.pascal.pascal_2012_converter.PascalSegConverter

Public Member Functions

ImageDataset convert (self)
 

Detailed Description

    Base class for Coretex Annotation format conversion

    Properties
    ----------
    datasetName : str
        name of dataset
    projectId : int
        id of Coretex Project
    datasetPath : str
        path to dataset

Definition at line 48 of file base_converter.py.

Member Function Documentation

◆ convert()

ImageDataset coretex.entities.conversion.base_converter.BaseConverter.convert (   self)
    Converts the dataset to Coretex Format

    Returns
    -------
    ImageDatasetType -> The converted ImageDataset object

Definition at line 86 of file base_converter.py.

86  def convert(self) -> ImageDataset:
87  """
88  Converts the dataset to Coretex Format
89 
90  Returns
91  -------
92  ImageDatasetType -> The converted ImageDataset object
93  """
94 
95  # Extract classes
96  labels = self._extractLabels()
97  classes = ImageDatasetClass.generate(labels)
98 
99  if self._dataset.saveClasses(classes):
100  logging.getLogger("coretexpylib").info(">> [Coretex] Dataset classes saved successfully")
101  else:
102  logging.getLogger("coretexpylib").info(">> [Coretex] Failed to save dataset classes")
103 
104  # Extract annotations
105  MultithreadedDataProcessor(
106  self._dataSource(),
107  self._extractSingleAnnotation,
108  message = "Converting dataset..."
109  ).process()
110 
111  if not self._dataset.finalize():
112  raise ValueError(f"Failed to finalize dataset \"{self._dataset.name}\"")
113 
114  return self._dataset

The documentation for this class was generated from the following file: