def find_dataset_by(self, dataset_id):
return DatasetModel.find_by(dataset_id)
def execute(self, dataset_id, dataset_source_folder_path):
dataset = DatasetModel.find_by(dataset_id)
self.move_content_from_source_to_dataset(dataset, dataset_source_folder_path)
self.image_repository.create_images_from(dataset.id)
return self.scan_annotation_from_json(dataset_id, dataset_source_folder_path)
def execute(self, dataset_id, categories):
dataset = DatasetModel.find_by(dataset_id)
category_id_list = CategoryModel.bulk_create(categories)
# dataset.update(set__categories=category_id_list)
dataset.update_categories(category_id_list)
def set_dataset_by(self, dataset_id):
self.dataset = DatasetModel.find_by(dataset_id)
