可视化的迁移¶
在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)]
|
要了解更多关于可视化的功能,请参阅这个教程。