YOLOv6在wandb下训练可视化

1、在tools下的train.py的下加入如下语句

import wandb
    
if __name__ == '__main__':
    wandb.init(project="项目名"，entity=“账户名”)
    # XXX
    # XXXX

2、在YOLOv6主目录下的yolov6/core/engine.py下添加如下语句

import wandb

class Trainer:
    def __init__(self, args, cfg, device):
        # XXX
        # XXX
        # XXX
        # XXX
        #最后一行
        wandb.config.update(dict(epoch=self.max_epoch,batch_size=self.batch_size))

3、 在YOLOv6主目录下的yolov6/utils/events.py添加如下语句

import wandb

def write_tblog(tblogger, epoch, results, lrs, losses):
    """Display mAP and loss information to log."""
    tblogger.add_scalar("val/mAP@0.5", results[0], epoch + 1)
    tblogger.add_scalar("val/mAP@0.50:0.95", results[1], epoch + 1)

    tblogger.add_scalar("train/iou_loss", losses[0], epoch + 1)
    tblogger.add_scalar("train/dist_focalloss", losses[1], epoch + 1)
    tblogger.add_scalar("train/cls_loss", losses[2], epoch + 1)

    tblogger.add_scalar("x/lr0", lrs[0], epoch + 1)
    tblogger.add_scalar("x/lr1", lrs[1], epoch + 1)
    tblogger.add_scalar("x/lr2", lrs[2], epoch + 1)

  
    wandb.log({"val/mAP@0.5":results[0],"epoch":epoch+1,
        "val/mAP@0.50:0.95":results[1],"train/iou_loss": losses[0], 
        "train/dist_focalloss":losses[1],
         "train/cls_loss":losses[2],
         "x/lr0":lrs[0],"x/lr1":lrs[1],
         "x/lr2":lrs[2]})