def __init__(self, path: PathService, dataset_cleaning: DatasetCleaningService, dataset_splitter: DatasetSplitterService, labels_to_csv_converter: LabelsToCsvConverterService, pbtxt_generator: PbtxtGeneratorService, tfrecords_generator: TfRecordGeneratorService): self.path: Paths = path.get_paths() self.dataset_cleaning = dataset_cleaning self.dataset_splitter = dataset_splitter self.labels_to_csv_converter = labels_to_csv_converter self.pbtxt_generator = pbtxt_generator self.tfrecords_generator = tfrecords_generator
def __init__(self, path: PathService, model_trainer: ModelTrainerService, model_eval: ModelEvaluationService): self.path: Paths = path.get_paths() self.model_train: ModelTrainerService = model_trainer self.model_eval: ModelEvaluationService = model_eval
def __init__(self, path: PathService): self.path: Paths = path.get_paths()
def __init__(self, path: PathService, docker_client: DockerClientService): self.path: Paths = path.get_paths() self.client: DockerClientService = docker_client.client
def __init__(self, path: PathService): self.path: Paths = path.get_paths() self.converter_instances: Dict[str, AbstractConverterService] = {} self.converter_mappings: Dict[str, AbstractConverterService] = {} self._initialize_mappings_()
def __init__(self, path: PathService): self.path: Paths = path.get_paths() self.label_map_dict: Dict[str, int] = {}