stylegan2_mmcv

build/lib/mmgen/.mim/configs/_base_/datasets/imagenet_128.py

+# dataset settings
+dataset_type = 'mmcls.ImageNet'
+
+# different from mmcls, we adopt the setting used in BigGAN.
+# We use `RandomCropLongEdge` in training and `CenterCropLongEdge` in testing.
+# Importantly, the `to_rgb` is set to `False` to remain image orders as BGR.
+img_norm_cfg = dict(
+    mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5], to_rgb=False)
+train_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='RandomCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(128, 128), backend='pillow'),
+    dict(type='RandomFlip', flip_prob=0.5, direction='horizontal'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='ToTensor', keys=['gt_label']),
+    dict(type='Collect', keys=['img', 'gt_label'])
+]
+
+test_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(128, 128), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='Collect', keys=['img'])
+]
+
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=2,
+    train=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/train',
+        pipeline=train_pipeline),
+    val=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline),
+    test=dict(
+        # replace `data/val` with `data/test` for standard test
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/imagenet_128x128_inception_stat.py

+# dataset settings
+dataset_type = 'mmcls.ImageNet'
+
+# This config is set for extract inception state of ImageNet dataset.
+# Following the pipeline of BigGAN, we center crop and resize images to 128x128
+# before feeding them to the Inception Net. Please refer to
+# https://github.com/ajbrock/BigGAN-PyTorch/blob/master/scripts/utils/prepare_data.sh
+# https://github.com/ajbrock/BigGAN-PyTorch/blob/master/make_hdf5.py
+# https://github.com/ajbrock/BigGAN-PyTorch/blob/master/calculate_inception_moments.py  # noqa
+
+# Importantly, the `to_rgb` is set to `True` to convert image orders to RGB.
+img_norm_cfg = dict(
+    mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5], to_rgb=True)
+train_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(128, 128), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='ToTensor', keys=['gt_label']),
+    dict(type='Collect', keys=['img', 'gt_label'])
+]
+
+test_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(128, 128), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='Collect', keys=['img'])
+]
+
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=2,
+    train=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/train',
+        pipeline=train_pipeline),
+    val=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline),
+    test=dict(
+        # replace `data/val` with `data/test` for standard test
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/imagenet_256.py

+# dataset settings
+dataset_type = 'mmcls.ImageNet'
+
+# different from mmcls, we adopt the setting used in BigGAN.
+# We use `RandomCropLongEdge` in training and `CenterCropLongEdge` in testing.
+# Importantly, the `to_rgb` is set to `False` to remain image orders as BGR.
+img_norm_cfg = dict(
+    mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5], to_rgb=False)
+train_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='RandomCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(256, 256), backend='pillow'),
+    dict(type='RandomFlip', flip_prob=0.5, direction='horizontal'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='ToTensor', keys=['gt_label']),
+    dict(type='Collect', keys=['img', 'gt_label'])
+]
+
+test_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(256, 256), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='Collect', keys=['img'])
+]
+
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=2,
+    train=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/train',
+        pipeline=train_pipeline),
+    val=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline),
+    test=dict(
+        # replace `data/val` with `data/test` for standard test
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/imagenet_64x64_inception_stat.py

+# dataset settings
+dataset_type = 'mmcls.ImageNet'
+
+# This config is set for extract inception state of ImageNet dataset.
+# Following the pipeline of BigGAN, we center crop and resize images to 128x128
+# before feeding them to the Inception Net. Please refer to
+# https://github.com/ajbrock/BigGAN-PyTorch/blob/master/scripts/utils/prepare_data.sh
+# https://github.com/ajbrock/BigGAN-PyTorch/blob/master/make_hdf5.py
+# https://github.com/ajbrock/BigGAN-PyTorch/blob/master/calculate_inception_moments.py  # noqa
+
+# Importantly, the `to_rgb` is set to `True` to convert image orders to RGB.
+img_norm_cfg = dict(
+    mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5], to_rgb=True)
+train_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(64, 64), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='ToTensor', keys=['gt_label']),
+    dict(type='Collect', keys=['img', 'gt_label'])
+]
+
+test_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(128, 128), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='Collect', keys=['img'])
+]
+
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=2,
+    train=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/train',
+        pipeline=train_pipeline),
+    val=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline),
+    test=dict(
+        # replace `data/val` with `data/test` for standard test
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/imagenet_noaug_128.py

+# dataset settings
+dataset_type = 'mmcls.ImageNet'
+
+# different from mmcls, we adopt the setting used in BigGAN.
+# Importantly, the `to_rgb` is set to `False` to remain image orders as BGR.
+# Remove `RandomFlip` augmentation and change `RandomCropLongEdge` to
+# `CenterCropLongEdge` to elminiate randomness.
+img_norm_cfg = dict(
+    mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5], to_rgb=False)
+train_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(128, 128), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='ToTensor', keys=['gt_label']),
+    dict(type='Collect', keys=['img', 'gt_label'])
+]
+
+test_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(128, 128), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='Collect', keys=['img'])
+]
+
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=2,
+    train=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/train',
+        pipeline=train_pipeline),
+    val=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline),
+    test=dict(
+        # replace `data/val` with `data/test` for standard test
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/imagenet_noaug_256.py

+# dataset settings
+dataset_type = 'mmcls.ImageNet'
+
+# different from mmcls, we adopt the setting used in BigGAN.
+# Importantly, the `to_rgb` is set to `False` to remain image orders as BGR.
+# Remove `RandomFlip` augmentation and change `RandomCropLongEdge` to
+# `CenterCropLongEdge` to elminiate randomness.
+img_norm_cfg = dict(
+    mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5], to_rgb=False)
+train_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(256, 256), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='ToTensor', keys=['gt_label']),
+    dict(type='Collect', keys=['img', 'gt_label'])
+]
+
+test_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(128, 128), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='Collect', keys=['img'])
+]
+
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=2,
+    train=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/train',
+        pipeline=train_pipeline),
+    val=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline),
+    test=dict(
+        # replace `data/val` with `data/test` for standard test
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/imagenet_noaug_64.py

+# dataset settings
+dataset_type = 'mmcls.ImageNet'
+
+# different from mmcls, we adopt the setting used in BigGAN.
+# Importantly, the `to_rgb` is set to `False` to remain image orders as BGR.
+# Remove `RandomFlip` augmentation and change `RandomCropLongEdge` to
+# `CenterCropLongEdge` to elminiate randomness.
+img_norm_cfg = dict(
+    mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5], to_rgb=False)
+train_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(64, 64), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='ToTensor', keys=['gt_label']),
+    dict(type='Collect', keys=['img', 'gt_label'])
+]
+
+test_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='CenterCropLongEdge', keys=['img']),
+    dict(type='Resize', size=(64, 64), backend='pillow'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='Collect', keys=['img'])
+]
+
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=2,
+    train=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/train',
+        pipeline=train_pipeline),
+    val=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline),
+    test=dict(
+        # replace `data/val` with `data/test` for standard test
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/imagenet_rgb.py

+# dataset settings
+dataset_type = 'mmcls.ImageNet'
+
+# Note that the pipelines below are from MMClassification
+img_norm_cfg = dict(
+    mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
+train_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='RandomResizedCrop', size=224, backend='pillow'),
+    dict(type='RandomFlip', flip_prob=0.5, direction='horizontal'),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='ToTensor', keys=['gt_label']),
+    dict(type='Collect', keys=['img', 'gt_label'])
+]
+
+test_pipeline = [
+    dict(type='LoadImageFromFile'),
+    dict(type='Resize', size=(256, -1), backend='pillow'),
+    dict(type='CenterCrop', crop_size=224),
+    dict(type='Normalize', **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img']),
+    dict(type='Collect', keys=['img'])
+]
+
+data = dict(
+    samples_per_gpu=64,
+    workers_per_gpu=2,
+    train=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/train',
+        pipeline=train_pipeline),
+    val=dict(
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline),
+    test=dict(
+        # replace `data/val` with `data/test` for standard test
+        type=dataset_type,
+        data_prefix='data/imagenet/val',
+        ann_file='data/imagenet/meta/val.txt',
+        pipeline=test_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/lsun-car_pad_512.py

+dataset_type = 'UnconditionalImageDataset'
+
+train_pipeline = [
+    dict(
+        type='LoadImageFromFile',
+        key='real_img',
+        io_backend='disk',
+    ),
+    dict(type='Resize', keys=['real_img'], scale=(512, 384)),
+    dict(
+        type='NumpyPad',
+        keys=['real_img'],
+        padding=((64, 64), (0, 0), (0, 0)),
+    ),
+    dict(type='Flip', keys=['real_img'], direction='horizontal'),
+    dict(
+        type='Normalize',
+        keys=['real_img'],
+        mean=[127.5] * 3,
+        std=[127.5] * 3,
+        to_rgb=False),
+    dict(type='ImageToTensor', keys=['real_img']),
+    dict(type='Collect', keys=['real_img'], meta_keys=['real_img_path'])
+]
+
+# `samples_per_gpu` and `imgs_root` need to be set.
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=4,
+    train=dict(
+        type='RepeatDataset',
+        times=5,
+        dataset=dict(
+            type=dataset_type, imgs_root=None, pipeline=train_pipeline)),
+    val=dict(type=dataset_type, imgs_root=None, pipeline=train_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/lsun_stylegan.py

+# Style-based GANs do not perform any augmentation for the LSUN datasets
+dataset_type = 'UnconditionalImageDataset'
+
+train_pipeline = [
+    dict(
+        type='LoadImageFromFile',
+        key='real_img',
+        io_backend='disk',
+    ),
+    dict(
+        type='Normalize',
+        keys=['real_img'],
+        mean=[127.5] * 3,
+        std=[127.5] * 3,
+        to_rgb=False),
+    dict(type='ImageToTensor', keys=['real_img']),
+    dict(type='Collect', keys=['real_img'], meta_keys=['real_img_path'])
+]
+
+val_pipeline = [
+    dict(
+        type='LoadImageFromFile',
+        key='real_img',
+        io_backend='disk',
+    ),
+    dict(
+        type='Normalize',
+        keys=['real_img'],
+        mean=[127.5] * 3,
+        std=[127.5] * 3,
+        to_rgb=True),
+    dict(type='ImageToTensor', keys=['real_img']),
+    dict(type='Collect', keys=['real_img'], meta_keys=['real_img_path'])
+]
+
+# `samples_per_gpu` and `imgs_root` need to be set.
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=4,
+    train=dict(
+        type='RepeatDataset',
+        times=100,
+        dataset=dict(
+            type=dataset_type, imgs_root=None, pipeline=train_pipeline)),
+    val=dict(type=dataset_type, imgs_root=None, pipeline=val_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/paired_imgs_256x256.py

+# dataset settings
+train_dataset_type = 'PairedImageDataset'
+val_dataset_type = 'PairedImageDataset'
+img_norm_cfg = dict(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
+train_pipeline = [
+    dict(
+        type='LoadPairedImageFromFile',
+        io_backend='disk',
+        key='pair',
+        flag='color'),
+    dict(
+        type='Resize',
+        keys=['img_a', 'img_b'],
+        scale=(256, 256),
+        interpolation='bicubic'),
+    dict(type='RescaleToZeroOne', keys=['img_a', 'img_b']),
+    dict(
+        type='Normalize', keys=['img_a', 'img_b'], to_rgb=True,
+        **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img_a', 'img_b']),
+    dict(
+        type='Collect',
+        keys=['img_a', 'img_b'],
+        meta_keys=['img_a_path', 'img_b_path'])
+]
+test_pipeline = [
+    dict(
+        type='LoadPairedImageFromFile',
+        io_backend='disk',
+        key='pair',
+        flag='color'),
+    dict(
+        type='Resize',
+        keys=['img_a', 'img_b'],
+        scale=(256, 256),
+        interpolation='bicubic'),
+    dict(type='RescaleToZeroOne', keys=['img_a', 'img_b']),
+    dict(
+        type='Normalize',
+        keys=['img_a', 'img_b'],
+        to_rgb=False,
+        **img_norm_cfg),
+    dict(type='ImageToTensor', keys=['img_a', 'img_b']),
+    dict(
+        type='Collect',
+        keys=['img_a', 'img_b'],
+        meta_keys=['img_a_path', 'img_b_path'])
+]
+
+data = dict(
+    samples_per_gpu=4,
+    workers_per_gpu=4,
+    drop_last=True,
+    train=dict(
+        type=train_dataset_type,
+        dataroot=None,
+        pipeline=train_pipeline,
+        test_mode=False),
+    val=dict(
+        type=val_dataset_type,
+        dataroot=None,
+        pipeline=test_pipeline,
+        test_mode=True),
+    test=dict(
+        type=val_dataset_type,
+        dataroot=None,
+        pipeline=test_pipeline,
+        test_mode=True))

build/lib/mmgen/.mim/configs/_base_/datasets/paired_imgs_256x256_crop.py

+# dataset settings
+train_dataset_type = 'PairedImageDataset'
+val_dataset_type = 'PairedImageDataset'
+img_norm_cfg = dict(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
+domain_a = 'photo'
+domain_b = 'mask'
+train_pipeline = [
+    dict(
+        type='LoadPairedImageFromFile',
+        io_backend='disk',
+        key='pair',
+        domain_a=domain_a,
+        domain_b=domain_b,
+        flag='color'),
+    dict(
+        type='Resize',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        scale=(286, 286),
+        interpolation='bicubic'),
+    dict(
+        type='FixedCrop',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        crop_size=(256, 256)),
+    dict(
+        type='Flip',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        direction='horizontal'),
+    dict(type='RescaleToZeroOne', keys=[f'img_{domain_a}', f'img_{domain_b}']),
+    dict(
+        type='Normalize',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        to_rgb=False,
+        **img_norm_cfg),
+    dict(type='ImageToTensor', keys=[f'img_{domain_a}', f'img_{domain_b}']),
+    dict(
+        type='Collect',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        meta_keys=[f'img_{domain_a}_path', f'img_{domain_b}_path'])
+]
+test_pipeline = [
+    dict(
+        type='LoadPairedImageFromFile',
+        io_backend='disk',
+        key='image',
+        domain_a=domain_a,
+        domain_b=domain_b,
+        flag='color'),
+    dict(
+        type='Resize',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        scale=(256, 256),
+        interpolation='bicubic'),
+    dict(type='RescaleToZeroOne', keys=[f'img_{domain_a}', f'img_{domain_b}']),
+    dict(
+        type='Normalize',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        to_rgb=False,
+        **img_norm_cfg),
+    dict(type='ImageToTensor', keys=[f'img_{domain_a}', f'img_{domain_b}']),
+    dict(
+        type='Collect',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        meta_keys=[f'img_{domain_a}_path', f'img_{domain_b}_path'])
+]
+
+data = dict(
+    samples_per_gpu=1,
+    workers_per_gpu=4,
+    drop_last=True,
+    train=dict(
+        type=train_dataset_type,
+        dataroot=None,
+        pipeline=train_pipeline,
+        test_mode=False),
+    val=dict(
+        type=val_dataset_type,
+        dataroot=None,
+        pipeline=test_pipeline,
+        test_mode=True),
+    test=dict(
+        type=val_dataset_type,
+        dataroot=None,
+        pipeline=test_pipeline,
+        test_mode=True))

build/lib/mmgen/.mim/configs/_base_/datasets/singan.py

+dataset_type = 'SinGANDataset'
+
+data = dict(
+    samples_per_gpu=1,
+    workers_per_gpu=4,
+    drop_last=False,
+    train=dict(
+        type=dataset_type,
+        img_path=None,  # need to set
+        min_size=25,
+        max_size=250,
+        scale_factor_init=0.75))

build/lib/mmgen/.mim/configs/_base_/datasets/unconditional_imgs_128x128.py

+dataset_type = 'UnconditionalImageDataset'
+
+train_pipeline = [
+    dict(type='LoadImageFromFile', key='real_img', io_backend='disk'),
+    dict(type='Resize', keys=['real_img'], scale=(128, 128)),
+    dict(
+        type='Normalize',
+        keys=['real_img'],
+        mean=[127.5] * 3,
+        std=[127.5] * 3,
+        to_rgb=False),
+    dict(type='ImageToTensor', keys=['real_img']),
+    dict(type='Collect', keys=['real_img'], meta_keys=['real_img_path'])
+]
+
+# `samples_per_gpu` and `imgs_root` need to be set.
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=4,
+    train=dict(type=dataset_type, imgs_root=None, pipeline=train_pipeline),
+    val=dict(type=dataset_type, imgs_root=None, pipeline=train_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/unconditional_imgs_64x64.py

+dataset_type = 'UnconditionalImageDataset'
+
+train_pipeline = [
+    dict(
+        type='LoadImageFromFile',
+        key='real_img',
+        io_backend='disk',
+    ),
+    dict(type='Resize', keys=['real_img'], scale=(64, 64)),
+    dict(
+        type='Normalize',
+        keys=['real_img'],
+        mean=[127.5] * 3,
+        std=[127.5] * 3,
+        to_rgb=False),
+    dict(type='ImageToTensor', keys=['real_img']),
+    dict(type='Collect', keys=['real_img'], meta_keys=['real_img_path'])
+]
+
+# `samples_per_gpu` and `imgs_root` need to be set.
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=4,
+    train=dict(type=dataset_type, imgs_root=None, pipeline=train_pipeline),
+    val=dict(type=dataset_type, imgs_root=None, pipeline=train_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/unconditional_imgs_flip_256x256.py

+dataset_type = 'UnconditionalImageDataset'
+
+train_pipeline = [
+    dict(type='LoadImageFromFile', key='real_img', io_backend='disk'),
+    dict(type='Resize', keys=['real_img'], scale=(256, 256)),
+    dict(type='Flip', keys=['real_img'], direction='horizontal'),
+    dict(
+        type='Normalize',
+        keys=['real_img'],
+        mean=[127.5] * 3,
+        std=[127.5] * 3,
+        to_rgb=False),
+    dict(type='ImageToTensor', keys=['real_img']),
+    dict(type='Collect', keys=['real_img'], meta_keys=['real_img_path'])
+]
+
+# `samples_per_gpu` and `imgs_root` need to be set.
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=4,
+    train=dict(type=dataset_type, imgs_root=None, pipeline=train_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/unconditional_imgs_flip_512x512.py

+dataset_type = 'UnconditionalImageDataset'
+
+train_pipeline = [
+    dict(type='LoadImageFromFile', key='real_img', io_backend='disk'),
+    dict(type='Resize', keys=['real_img'], scale=(512, 512)),
+    dict(type='Flip', keys=['real_img'], direction='horizontal'),
+    dict(
+        type='Normalize',
+        keys=['real_img'],
+        mean=[127.5] * 3,
+        std=[127.5] * 3,
+        to_rgb=False),
+    dict(type='ImageToTensor', keys=['real_img']),
+    dict(type='Collect', keys=['real_img'], meta_keys=['real_img_path'])
+]
+
+# `samples_per_gpu` and `imgs_root` need to be set.
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=4,
+    train=dict(type=dataset_type, imgs_root=None, pipeline=train_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/unconditional_imgs_flip_lanczos_resize_256x256.py

+dataset_type = 'UnconditionalImageDataset'
+
+train_pipeline = [
+    dict(type='LoadImageFromFile', key='real_img'),
+    dict(
+        type='Resize',
+        keys=['real_img'],
+        scale=(256, 256),
+        interpolation='lanczos',
+        backend='pillow'),
+    dict(type='Flip', keys=['real_img'], direction='horizontal'),
+    dict(
+        type='Normalize',
+        keys=['real_img'],
+        mean=[127.5] * 3,
+        std=[127.5] * 3,
+        to_rgb=False),
+    dict(type='ImageToTensor', keys=['real_img']),
+    dict(type='Collect', keys=['real_img'], meta_keys=['real_img_path'])
+]
+
+# `samples_per_gpu` and `imgs_root` need to be set.
+data = dict(
+    samples_per_gpu=None,
+    workers_per_gpu=4,
+    train=dict(
+        type='RepeatDataset',
+        times=100,
+        dataset=dict(
+            type=dataset_type, imgs_root=None, pipeline=train_pipeline)),
+    val=dict(type=dataset_type, imgs_root=None, pipeline=train_pipeline))

build/lib/mmgen/.mim/configs/_base_/datasets/unpaired_imgs_256x256.py

+train_dataset_type = 'UnpairedImageDataset'
+val_dataset_type = 'UnpairedImageDataset'
+img_norm_cfg = dict(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
+domain_a = None  # set by user
+domain_b = None  # set by user
+train_pipeline = [
+    dict(
+        type='LoadImageFromFile',
+        io_backend='disk',
+        key=f'img_{domain_a}',
+        flag='color'),
+    dict(
+        type='LoadImageFromFile',
+        io_backend='disk',
+        key=f'img_{domain_b}',
+        flag='color'),
+    dict(
+        type='Resize',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        scale=(286, 286),
+        interpolation='bicubic'),
+    dict(
+        type='Crop',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        crop_size=(256, 256),
+        random_crop=True),
+    dict(type='Flip', keys=[f'img_{domain_a}'], direction='horizontal'),
+    dict(type='Flip', keys=[f'img_{domain_b}'], direction='horizontal'),
+    dict(type='RescaleToZeroOne', keys=[f'img_{domain_a}', f'img_{domain_b}']),
+    dict(
+        type='Normalize',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        to_rgb=False,
+        mean=[0.5, 0.5, 0.5],
+        std=[0.5, 0.5, 0.5]),
+    dict(type='ImageToTensor', keys=[f'img_{domain_a}', f'img_{domain_b}']),
+    dict(
+        type='Collect',
+        keys=[f'img_{domain_a}', f'img_{domain_b}'],
+        meta_keys=[f'img_{domain_a}_path', f'img_{domain_b}_path'])
+]
+test_pipeline = [
+    dict(
+        type='LoadImageFromFile', io_backend='disk', key='image',
+        flag='color'),
+    dict(
+        type='Resize',
+        keys=['image'],
+        scale=(256, 256),
+        interpolation='bicubic'),
+    dict(type='RescaleToZeroOne', keys=['image']),
+    dict(
+        type='Normalize',
+        keys=['image'],
+        to_rgb=False,
+        mean=[0.5, 0.5, 0.5],
+        std=[0.5, 0.5, 0.5]),
+    dict(type='ImageToTensor', keys=['image']),
+    dict(type='Collect', keys=['image'], meta_keys=['image_path'])
+]
+data_root = None
+data = dict(
+    samples_per_gpu=1,
+    workers_per_gpu=4,
+    drop_last=True,
+    train=dict(
+        type=train_dataset_type,
+        dataroot=data_root,
+        pipeline=train_pipeline,
+        test_mode=False,
+        domain_a=domain_a,
+        domain_b=domain_b),
+    val=dict(
+        type=val_dataset_type,
+        dataroot=data_root,
+        pipeline=test_pipeline,
+        test_mode=True,
+        domain_a=domain_a,
+        domain_b=domain_b),
+    test=dict(
+        type=val_dataset_type,
+        dataroot=data_root,
+        pipeline=test_pipeline,
+        test_mode=True,
+        domain_a=domain_a,
+        domain_b=domain_b))

build/lib/mmgen/.mim/configs/_base_/default_metrics.py

+metrics = dict(
+    fid50k=dict(type='FID', num_images=50000),
+    pr50k3=dict(type='PR', num_images=50000, k=3),
+    is50k=dict(type='IS', num_images=50000),
+    ppl_zfull=dict(type='PPL', space='Z', sampling='full', num_images=50000),
+    ppl_wfull=dict(type='PPL', space='W', sampling='full', num_images=50000),
+    ppl_zend=dict(type='PPL', space='Z', sampling='end', num_images=50000),
+    ppl_wend=dict(type='PPL', space='W', sampling='end', num_images=50000),
+    ms_ssim10k=dict(type='MS_SSIM', num_images=10000),
+    swd16k=dict(type='SWD', num_images=16384))