labels.jpg names fix (#9361)

Partially resolves https://github.com/ultralytics/yolov5/issues/9360

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-09-10 22:24:46 +03:00
1 parent 4e8504abd9
commit 2b5c9a83ec
1 file changed
+1 -1
+1 -1
View File
@@ -364,7 +364,7 @@ def plot_labels(labels, names=(), save_dir=Path('')):
ax[0].set_ylabel('instances')
if 0 < len(names) < 30:
ax[0].set_xticks(range(len(names)))
ax[0].set_xticklabels(names, rotation=90, fontsize=10)
ax[0].set_xticklabels(list(names.values()), rotation=90, fontsize=10)
else:
ax[0].set_xlabel('classes')
sn.histplot(x, x='x', y='y', ax=ax[2], bins=50, pmax=0.9)