Bases: ABC
Base converter class. Based on the source and destination formats, selects reader and writer classes for
processing data
Source code in tools/annotation_converter/converter/base.py
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46 | class BaseConverter(ABC):
"""
Base converter class. Based on the source and destination formats, selects reader and writer classes for
processing data
"""
def __init__(self, log_level: str = LevelMapping.debug, log_path: Optional[Path] = None):
self._reader: Optional[BaseReader] = None
self._writer: Optional[YoloWriter] = None
self.reader_mapping = {
".xml": XMLReader,
}
self.writer_mapping = {
".txt": YoloWriter,
}
self.logger = LoggerConfigurator.setup(
name=self.__class__.__name__,
log_level=log_level,
log_path=Path(log_path) / f"{self.__class__.__name__}.log" if log_path else None
)
@abstractmethod
def convert(self, file_paths: Tuple[Path], target_path: Path, n_jobs: int = 1) -> None:
pass
@property
def reader(self) -> BaseReader:
return self._reader
@reader.setter
def reader(self, reader: BaseReader) -> None:
self._reader = reader
|