mmagic.datasets.dreambooth_dataset

Module Contents

Classes

DreamBoothDataset

Dataset for DreamBooth.

class mmagic.datasets.dreambooth_dataset.DreamBoothDataset(data_root: str, concept_dir: str, prompt: str, pipeline: List[Union[dict, Callable]] = [])

Bases: mmengine.dataset.BaseDataset

Dataset for DreamBooth.

Parameters

• data_root (str) – Path to the data root.

• concept_dir (str) – Path to the concept images.

• prompt (str) – Prompt of the concept.

• pipeline (list[dict | callable]) – A sequence of data transforms.

load_data_list() → list

Load data list from concept_dir and class_dir.