wandb(Weights & Biases)是一个类似于tensorboard的在线模型训练可视化工具。

1)注册和安装wandb

注册wandb
到其官网 https://wandb.ai/home 注册
安装wandb
执行:

pip install wandb

在这里插入图片描述

wandb login

这个输入的时候,貌似不能paste,只能一个个地手动输入
注意API 输入的是,登录wandb后,个人界面里有一个API keys
在这里插入图片描述

在这里插入图片描述

2)安装wandb完之后

可以在C盘user目录中找到.netrc文件,里面有你的登录密码
在这里插入图片描述

3) 在代码中修改项目名称

在utils/wandb_logging/wandb_utils.py中,把project名字改成你的项目名

 self.wandb_run = wandb.init(config=opt,
                                    resume="allow",
                                    project='YOLOv5-Ball' if opt.project == 
'runs/train' else Path(opt.project).stem,
                                    name=name,
                                    job_type=job_type,
                                    id=run_id) if not wandb.run else wandb.run
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐