Metric-Confidence plots feature addition (#2057)
* Metric-Confidence plots feature addition * cleanup * Metric-Confidence plots feature addition * cleanup * Update run-once lines * cleanup * save all 4 curves to wandb
This commit is contained in:
1 parent
2a835c79a9
commit
f639e14e4d
3 files changed
+40
-17
No files matched your search
@@ -403,7 +403,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
|
||||
if plots:
|
||||
plot_results(save_dir=save_dir) # save as results.png
|
||||
if wandb:
|
||||
files = ['results.png', 'precision_recall_curve.png', 'confusion_matrix.png']
|
||||
files = ['results.png', 'confusion_matrix.png', *[f'{x}_curve.png' for x in ('F1', 'PR', 'P', 'R')]]
|
||||
wandb.log({"Results": [wandb.Image(str(save_dir / f), caption=f) for f in files
|
||||
if (save_dir / f).exists()]})
|
||||
if opt.log_artifacts:
|
||||
|
||||
Reference in new issue
Block a user