训练过程可视化有两种方式:wandb和tensorboard,这里我采用的是wandb,想要在训练过程中调用wandb只需要在要训练的配置文件(如yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco.py)中加上一行代码即可:
visualizer = dict(vis_backends = [dict(type='LocalVisBackend'), dict(type='WandbVisBackend')])
当然想要使用wandb还需要安装wandb:pip install wandb
关于wandb更多的使用方法大家可以看其他博客,本文章只是记录下如何在yolo-world训练过程中加入可视化过程。
注:如果采用tensorboard的话可以加上这个代码:
visualizer = dict(vis_backends=[dict(type='LocalVisBackend'),dict(type='TensorboardVisBackend')])