Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

One yolov5 support wandb train #87

Merged
merged 2 commits into from Nov 30, 2022
Merged

One yolov5 support wandb train #87

merged 2 commits into from Nov 30, 2022

Conversation

ccssu
Copy link
Collaborator

@ccssu ccssu commented Nov 30, 2022

https://wandb.ai/home
此pr修复 项目中 wandb 对实验跟踪和可视化功能的支持。

使用coco128数据集 对 wandb 集成可视化结果: https://wandb.ai/wearmheart/YOLOv5/runs/3si719qd?workspace=user-wearmheart

简单二步即可开始记录机器学习实验。

1. 安装库

pip install wandb

2. 创建账号

注册页注册一个免费账号

image

终端输入

wandb login

终端输入后粘贴copy的key 输入回车确认 ,大工告成。

验证

使用coco128数据集 对 wandb 集成可视化结果: https://wandb.ai/wearmheart/YOLOv5/runs/3si719qd?workspace=user-wearmheart

使用指令 python train.py --weights ' ' --data data/coco128.yaml --cfg models/yolov5s.yaml
成功运行示例如下:
image

通过wandb: 🚀 View run at:xxx链接即可查看 使用wandb可视化的结果。

Copy link
Contributor

@BBuf BBuf left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

good job! thanks @wyg1997

@BBuf BBuf merged commit 82e2e05 into main Nov 30, 2022
@BBuf BBuf deleted the one-yolov5-support-wandb-train branch December 4, 2022 01:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants