跳转至

可视化的迁移⚓︎

在0.x版中，MMEditing使用VisualizationHook来对训练过程中生成的结果进行可视化，在1.x版中，我们将该功能整合到BasicVisualizationHook / VisualizationHook中，而且遵循MMEngine的设计，我们实现了ConcatImageVisualizer / Visualizer和一系列VisBackend来绘制和保存可视化结果。

0.x版 1.x版 
visual_config = dict(
    type='VisualizationHook',
    output_dir='visual',
    interval=1000,
    res_name_list=['gt_img', 'masked_img', 'fake_res', 'fake_img'],
)

vis_backends = [dict(type='LocalVisBackend')]
visualizer = dict(
    type='ConcatImageVisualizer',
    vis_backends=vis_backends,
    fn_key='gt_path',
    img_keys=['gt_img', 'input', 'pred_img'],
    bgr2rgb=True)
custom_hooks = [dict(type='BasicVisualizationHook', interval=1)]

要了解更多关于可视化的功能，请参阅这个教程。

最后更新: November 27, 2023
创建日期: November 27, 2023