Why?
当我们训练模型时(有监督),对训练集、验证集的数据分布需要有一个大致的概念。
样本空间是否均衡?
训练集和验证集比例是否平衡?
有哪些标签类别?
每类标签数量如何?
采用yolo系列训练时,模型训练完成时,当然会生成一张统计表,但是不够美观和直观。
特别是向leader汇报工作时,我们更希望能够以图表的形式展示出来。
所以,上图的结果,如果能够加上标签分布统计图,就能够起到锦上添花的效果:
How?代码文件附在文末
本文采用jupyter作为简单的展示,代码参考下图: