mmagic.datasets.comp1k_dataset
¶
Module Contents¶
Classes¶
Adobe composition-1k dataset. |
- class mmagic.datasets.comp1k_dataset.AdobeComp1kDataset(ann_file: Optional[str] = '', metainfo: Union[collections.abc.Mapping, mmengine.config.Config, None] = None, data_root: Optional[str] = '', data_prefix: dict = dict(img_path=''), filter_cfg: Optional[dict] = None, indices: Optional[Union[int, Sequence[int]]] = None, serialize_data: bool = True, pipeline: List[Union[dict, Callable]] = [], test_mode: bool = False, lazy_init: bool = False, max_refetch: int = 1000)[源代码]¶
Bases:
mmengine.dataset.BaseDataset
Adobe composition-1k dataset.
The dataset loads (alpha, fg, bg) data and apply specified transforms to the data. You could specify whether composite merged image online or load composited merged image in pipeline.
Example for online comp-1k dataset:
[ { "alpha": 'alpha/000.png', "fg": 'fg/000.png', "bg": 'bg/000.png' }, { "alpha": 'alpha/001.png', "fg": 'fg/001.png', "bg": 'bg/001.png' }, ]
Example for offline comp-1k dataset:
[ { "alpha": 'alpha/000.png', "merged": 'merged/000.png', "fg": 'fg/000.png', "bg": 'bg/000.png' }, { "alpha": 'alpha/001.png', "merged": 'merged/001.png', "fg": 'fg/001.png', "bg": 'bg/001.png' }, ]
- 参数
ann_file (str) – Annotation file path. Defaults to ‘’.
data_root (str, optional) – The root directory for
data_prefix
andann_file
. Defaults to None.pipeline (list, optional) – Processing pipeline. Defaults to [].
test_mode (bool, optional) –
test_mode=True
means in test phase. Defaults to False.**kwargs – Other arguments passed to
mmengine.dataset.BaseDataset
.
实际案例
See unit-tests TODO: Move some codes in unittest here