Mmdetection: How to add images to tensorboard during training?

[wangg12]

[OceanPang]

@wangg12 mmdetection use hooks to control those modules. If you want to show images on tensorboard, you should first add the images into Runner object (See in mmcv), and upload them to tensorboard using tensorboard hooks

[wangg12]

@OceanPang Thank you. It works.

[LUCKMOONLIGHT]

@OceanPang Thank you. It works.
How did you do it, can you share with me?

[UpCoder]

@LUCKMOONLIGHT I find a way

1. You need to rewrite the Tensorboard Hook, such as TensorboardImageHook
2. You need to pass the image object in batch_processor func in train.py to the outputs
3. You can write the image object to tensorboard in your TensorboardImageHook