Add writer (#74)

* add writer

* update

* update

* update docstr

* update unittest

* update unittest

* fix comment

* update unittest

* fix comment

* fix comment

* fix comment

* fix comment

* update

mmengine/registry/__init__.py

@@ -3,11 +3,11 @@ from .registry import Registry, build_from_cfg
 from .root import (DATA_SAMPLERS, DATASETS, EVALUATORS, HOOKS, MODEL_WRAPPERS,
                    MODELS, OPTIMIZER_CONSTRUCTORS, OPTIMIZERS,
                    PARAM_SCHEDULERS, RUNNER_CONSTRUCTORS, RUNNERS, TASK_UTILS,
-                   TRANSFORMS, VISUALIZERS, WEIGHT_INITIALIZERS)
+                   TRANSFORMS, VISUALIZERS, WEIGHT_INITIALIZERS, WRITERS)
 
 __all__ = [
     'Registry', 'build_from_cfg', 'RUNNERS', 'RUNNER_CONSTRUCTORS', 'HOOKS',
     'DATASETS', 'DATA_SAMPLERS', 'TRANSFORMS', 'MODELS', 'WEIGHT_INITIALIZERS',
     'OPTIMIZERS', 'OPTIMIZER_CONSTRUCTORS', 'TASK_UTILS', 'PARAM_SCHEDULERS',
-    'EVALUATORS', 'MODEL_WRAPPERS', 'VISUALIZERS'
+    'EVALUATORS', 'MODEL_WRAPPERS', 'WRITERS', 'VISUALIZERS'
 ]

mmengine/registry/root.py

@@ -39,6 +39,7 @@ TASK_UTILS = Registry('task util')
 
 # manage all kinds of evaluators for computing metrics
 EVALUATORS = Registry('evaluator')
-
 # manage visualizer
 VISUALIZERS = Registry('visualizer')
+# manage writer
+WRITERS = Registry('writer')

mmengine/visualization/__init__.py

 # Copyright (c) OpenMMLab. All rights reserved.
 from .visualizer import Visualizer
+from .writer import (BaseWriter, ComposedWriter, LocalWriter,
+                     TensorboardWriter, WandbWriter)
 
-__all__ = ['Visualizer']
+__all__ = [
+    'Visualizer', 'BaseWriter', 'LocalWriter', 'WandbWriter',
+    'TensorboardWriter', 'ComposedWriter'
+]