Shortcuts

数据集验证

在 MMPretrain 中,tools/misc/verify_dataset.py 脚本会检查数据集的所有图片,查看是否有已经损坏的图片。

工具介绍

python tools/print_config.py \
    ${CONFIG} \
    [--out-path ${OUT-PATH}] \
    [--phase ${PHASE}] \
    [--num-process ${NUM-PROCESS}]
    [--cfg-options ${CFG_OPTIONS}]

所有参数说明:

  • config : 配置文件的路径。

  • --out-path : 输出结果路径,默认为 ‘brokenfiles.log’。

  • --phase : 检查哪个阶段的数据集,可用值为 “train” 、”test” 或者 “val”, 默认为 “train”。

  • --num-process : 指定的进程数,默认为 1。

  • --cfg-options: 额外的配置选项,会被合入配置文件,参考教程 1:如何编写配置文件

示例:

python tools/misc/verify_dataset.py configs/t2t_vit/t2t-vit-t-14_8xb64_in1k.py --out-path broken_imgs.log --phase val --num-process 8
Read the Docs v: latest
Versions
latest
stable
mmcls-1.x
mmcls-0.x
dev
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.