Ejemplo n.º 1
0
 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
Ejemplo n.º 2
0
 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
Ejemplo n.º 3
0
 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
Ejemplo n.º 5
0
 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_()
Ejemplo n.º 6
0
 def __init__(self, path: PathService):
     self.path: Paths = path.get_paths()
     self.label_map_dict: Dict[str, int] = {}