Source code for matting_default_runtime

# Copyright (c) OpenMMLab. All rights reserved.
from mmengine.hooks import (CheckpointHook, DistSamplerSeedHook, IterTimerHook,
                            LoggerHook, ParamSchedulerHook)
from mmengine.runner import LogProcessor
from mmengine.visualization import LocalVisBackend

from mmagic.engine.hooks import BasicVisualizationHook
from mmagic.visualization import ConcatImageVisualizer

[docs]default_scope = 'mmagic'
[docs]save_dir = './work_dirs'

[docs]default_hooks = dict(
    timer=dict(type=IterTimerHook),
    logger=dict(type=LoggerHook, interval=100),
    param_scheduler=dict(type=ParamSchedulerHook),
    checkpoint=dict(
        type=CheckpointHook, interval=10000, by_epoch=False, out_dir=save_dir),
    sampler_seed=dict(type=DistSamplerSeedHook),
)

[docs]env_cfg = dict(
    cudnn_benchmark=False,
    mp_cfg=dict(mp_start_method='fork', opencv_num_threads=4),
    dist_cfg=dict(backend='nccl'),
)

[docs]vis_backends = [dict(type=LocalVisBackend)]
[docs]visualizer = dict(
    type=ConcatImageVisualizer,
    vis_backends=vis_backends,
    fn_key='trimap_path',
    img_keys=['pred_alpha', 'trimap', 'gt_merged', 'gt_alpha'],
    bgr2rgb=True)
[docs]custom_hooks = [dict(type=BasicVisualizationHook, interval=2000)]

[docs]log_level = 'INFO'
[docs]log_processor = dict(type=LogProcessor, by_epoch=False)

[docs]load_from = None
[docs]resume = False

# TODO: support auto scaling lr