MMCV - browse_dataset.py 可视化检测和跟踪任务config文件数据预处理部分神器

刚哥吧

已于 2023-02-01 14:08:21 修改

阅读量1.2k

点赞数 1

分类专栏：目标跟踪干货分享文章标签： MMTracking 数据可视化 MOT browse_dataset json

于 2023-01-31 22:34:27 首次发布

目标跟踪干货分享专栏收录该内容

15 篇文章 ¥299.90 ¥399.90

订阅专栏

本文介绍了如何利用MMCV框架提供的browse_dataset.py工具，对mmdetection、mmtracking等项目的配置文件中的数据预处理部分进行可视化检查。通过设置不同参数，可以观察原始图片、预处理后的图片或数据流水线的中间过程，帮助验证标注文件的正确性和数据处理的效果。

无论是mmdetection、mmtracking、mmdetection3D等框架，在\tools\analysis_tools中均有一个名为browse_dataset.py的文件。该文件能够帮助用户直接可视化 config 文件中的数据处理部分，查看标注文件是否正确，同时可以选择保存可视化图片到指定文件夹内。

调用的基本命令如下：

python tools/analysis_tools/browse_dataset.py \
    ${CONFIG_FILE} \
    [-o, --output-dir ${OUTPUT_DIR}] \
    [-p, --phase ${DATASET_PHASE}] \
    [-n, --show-number ${NUMBER_IMAGES_DISPLAY}] \
    [-i, --show-interval ${SHOW_INTERRVAL}] \
    [-m, --mode ${DISPLAY_MODE}] \
    [--cfg-options ${CFG_OPTIONS}]

参数说明：