做视觉检测跟踪任务时,我们需要在论文插图中体现出我们数据集的信息,这个时候就有一个非常好用的神器:dataset_analysis.py的文件。该文件能够帮助用户直接可视化custom数据集的bbox实例信息,如上图所示,包括:显示类别和 bbox 实例个数的分布图;显示类别和 bbox 实例宽/高比例的分布图;显示类别和 bbox 实例个数的分布图;显示类别和 bbox 实例面积的分布图。该文件的位置为:C:\Users\dingjiangang\Desktop\mmd_all\mmyolo\mmyolo-0.4.0\tools\analysis_tools\dataset_analysis.py (mmdetection和mmtracking等框架中不含,只有mmyolo才有)。
该脚本只需要传入参数config文件即可,通过获取dataloder从而可视化数据:</