labels.jpg class names (#2454)

* labels.png class names

* fontsize=10
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-03-12 22:15:41 -08:00
1 parent f4197214aa
commit 08d4918d7f
2 files changed
+8 -3

No files matched your search

+1 -1
View File
@@ -203,7 +203,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
# cf = torch.bincount(c.long(), minlength=nc) + 1. # frequency
# model._initialize_biases(cf.to(device))
if plots:
plot_labels(labels, save_dir, loggers)
plot_labels(labels, names, save_dir, loggers)
if tb_writer:
tb_writer.add_histogram('classes', c, 0)