Table of Contents
Shortcuts

mmedit.datasets.dreambooth_dataset

Module Contents

Classes

DreamBoothDataset

Dataset for DreamBooth.
class mmedit.datasets.dreambooth_dataset.DreamBoothDataset(data_root: str, concept_dir: str, prompt: str, pipeline: List[Union[dict, Callable]] = [])[源代码]

Bases: mmengine.dataset.BaseDataset

Dataset for DreamBooth.

参数

  • 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.