Optimised Callback Class to Reduce Code and Fix Errors (#4688)
* added callbacks * added back callback to main * added save_dir to callback output * reduced code count * updated callbacks * added default callback class to main, added missing parameters to on_model_save * Glenn updates Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
548745181a
commit
2317f86ca4
3 files changed
+23
-126
No files matched your search
@@ -216,7 +216,7 @@ def run(data,
|
||||
save_one_txt(predn, save_conf, shape, file=save_dir / 'labels' / (path.stem + '.txt'))
|
||||
if save_json:
|
||||
save_one_json(predn, jdict, path, class_map) # append to COCO-JSON dictionary
|
||||
callbacks.on_val_image_end(pred, predn, path, names, img[si])
|
||||
callbacks.run('on_val_image_end', pred, predn, path, names, img[si])
|
||||
|
||||
# Plot images
|
||||
if plots and batch_i < 3:
|
||||
@@ -253,7 +253,7 @@ def run(data,
|
||||
# Plots
|
||||
if plots:
|
||||
confusion_matrix.plot(save_dir=save_dir, names=list(names.values()))
|
||||
callbacks.on_val_end()
|
||||
callbacks.run('on_val_end')
|
||||
|
||||
# Save JSON
|
||||
if save_json and len(jdict):
|
||||
|
||||
Reference in new issue
Block a user