MMCV - dataset_analysis.py 可视化检测和跟踪任务自定义数据集神器

MMCV - dataset_analysis.py 可视化检测和跟踪任务自定义数据集神器_第1张图片

做视觉检测跟踪任务时,我们需要在论文插图中体现出我们数据集的信息,这个时候就有一个非常好用的神器: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从而可视化数据:

你可能感兴趣的:(目标跟踪干货分享,数据可视化,mmcv,mmdetection,mmyolo)