diff --git a/mmengine/visualization/vis_backend.py b/mmengine/visualization/vis_backend.py index 73b06fa26a981376043b3187ae264b77a3aa123d..1f894d43a77b05295f98fe918b10c74c41b01e64 100644 --- a/mmengine/visualization/vis_backend.py +++ b/mmengine/visualization/vis_backend.py @@ -411,6 +411,7 @@ class WandbVisBackend(BaseVisBackend): step (int): Useless parameter. Wandb does not need this parameter. Default to 0. """ + image = self._wandb.Image(image) self._wandb.log({name: image}, commit=self._commit) @force_init_env