* W&B log epoch * capitalize * W&B log epoch * capitalize * Update train.py New try using https://docs.wandb.ai/library/log#incremental-logging * Update train.py * Update test.py * Update train.py * Update plots.py * Update train.py * Update train.py * label plot step -1 * update * update * update * update * update * update * Update train.py * Update train.py5.0
if plots: | if plots: | ||||
confusion_matrix.plot(save_dir=save_dir, names=list(names.values())) | confusion_matrix.plot(save_dir=save_dir, names=list(names.values())) | ||||
if wandb and wandb.run: | if wandb and wandb.run: | ||||
wandb.log({"Images": wandb_images}) | |||||
wandb.log({"Validation": [wandb.Image(str(f), caption=f.name) for f in sorted(save_dir.glob('test*.jpg'))]}) | |||||
val_batches = [wandb.Image(str(f), caption=f.name) for f in sorted(save_dir.glob('test*.jpg'))] | |||||
wandb.log({"Images": wandb_images, "Validation": val_batches}, commit=False) | |||||
# Save JSON | # Save JSON | ||||
if save_json and len(jdict): | if save_json and len(jdict): |
# tb_writer.add_graph(model, imgs) # add model to tensorboard | # tb_writer.add_graph(model, imgs) # add model to tensorboard | ||||
elif plots and ni == 10 and wandb: | elif plots and ni == 10 and wandb: | ||||
wandb.log({"Mosaics": [wandb.Image(str(x), caption=x.name) for x in save_dir.glob('train*.jpg') | wandb.log({"Mosaics": [wandb.Image(str(x), caption=x.name) for x in save_dir.glob('train*.jpg') | ||||
if x.exists()]}) | |||||
if x.exists()]}, commit=False) | |||||
# end batch ------------------------------------------------------------------------------------------------ | # end batch ------------------------------------------------------------------------------------------------ | ||||
# end epoch ---------------------------------------------------------------------------------------------------- | # end epoch ---------------------------------------------------------------------------------------------------- |
# loggers | # loggers | ||||
for k, v in loggers.items() or {}: | for k, v in loggers.items() or {}: | ||||
if k == 'wandb' and v: | if k == 'wandb' and v: | ||||
v.log({"Labels": [v.Image(str(x), caption=x.name) for x in save_dir.glob('*labels*.jpg')]}) | |||||
v.log({"Labels": [v.Image(str(x), caption=x.name) for x in save_dir.glob('*labels*.jpg')]}, commit=False) | |||||
def plot_evolution(yaml_file='data/hyp.finetune.yaml'): # from utils.plots import *; plot_evolution() | def plot_evolution(yaml_file='data/hyp.finetune.yaml'): # from utils.plots import *; plot_evolution() |