W&B feature improvements (#1258)
* W&B feature improvements This PR add: * Class to id labels. Now, the caption of bounding boxes will display the class name and the class confidence score. * The project name is set to "Yolov5" and the run name will be set to opt.logdir * cleanup * remove parenthesis on caption Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
2062765e06
commit
96fcde40b8
2 files changed
+9
-7
No files matched your search
@@ -121,7 +121,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
|
||||
# Logging
|
||||
if wandb and wandb.run is None:
|
||||
id = ckpt.get('wandb_id') if 'ckpt' in locals() else None
|
||||
wandb_run = wandb.init(config=opt, resume="allow", project=os.path.basename(log_dir), id=id)
|
||||
wandb_run = wandb.init(config=opt, resume="allow", project="YOLOv5", name=os.path.basename(log_dir), id=id)
|
||||
|
||||
# Resume
|
||||
start_epoch, best_fitness = 0, 0.0
|
||||
|
||||
Reference in new issue
Block a user