Skip to content

[Improvement] browse_dataset.py#304

Merged
hhaAndroid merged 13 commits into
open-mmlab:devfrom
matrixgame2018:update-tools/browse_dataset
Dec 1, 2022
Merged

[Improvement] browse_dataset.py#304
hhaAndroid merged 13 commits into
open-mmlab:devfrom
matrixgame2018:update-tools/browse_dataset

Conversation

@matrixgame2018

@matrixgame2018 matrixgame2018 commented Nov 20, 2022

Copy link
Copy Markdown
Collaborator

浏览数据集

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}] \
    [-r, --rescale-factor ${RESCALE_FACTOR}] \
    [-c, --channel-order ${CHANNEL_ORDER}] \
    [--cfg-options ${CFG_OPTIONS}]

所有参数的说明

  • config : 模型配置文件的路径。
  • -o, --output-dir: 保存图片文件夹,如果没有指定,默认为 '',表示不保存。
  • -p, --phase: 可视化数据集的阶段,只能为 ['train', 'val', 'test'] 之一,默认为 'train'
  • -n, --show-number: 可视化样本数量。如果没有指定,默认展示数据集的所有图片。
  • -i, --show-interval: 浏览时,每张图片的停留间隔,单位为秒。
  • -m, --mode: 可视化的模式,只能为 ['original', 'transformed', 'concat', 'pipeline'] 之一。 默认为'transformed'.
  • --cfg-options : 对配置文件的修改,参考[学习配置文件](./config.md)。
`-m, --mode` 用于设置可视化的模式,默认设置为 'transformed'。
- 如果 `--mode` 设置为 'original',则获取原始图片;
- 如果 `--mode` 设置为 'transformed',则获取预处理后的图片;
- 如果 `--mode` 设置为 'concat',获取原始图片和预处理后图片拼接的图片;
- 如果 `--mode` 设置为 'pipeline',则获得数据流水线所有中间过程图片。

示例

  1. 'original' 模式 :
python ./tools/analysis_tools/browse_dataset.py configs/yolov5/yolov5_balloon.py --phase val --output-dir tmp --mode original
  • --phase val: 可视化验证集, 可简化为 -p val;
  • --output-dir tmp: 可视化结果保存在 "tmp" 文件夹, 可简化为 -o tmp;
  • --mode original: 可视化原图, 可简化为 -m original;
  • --show-number 100: 可视化100张图,可简化为 -n 100;
  1. 'transformed' 模式 :
python ./tools/analysis_tools/browse_dataset.py configs/yolov5/yolov5_balloon.py 
  1. 'concat' 模式 :
python ./tools/analysis_tools/browse_dataset.py configs/yolov5/yolov5_balloon.py -m concat
  1. 'pipeline' 模式 :
python ./tools/analysis_tools/browse_dataset.py configs/yolov5/yolov5_balloon.py -m pipeline

Upgrate the brow_transform
对pipeline中的transform过程实现了可视化,以及可以将transform中key参数的变化打印出来。
@PeterH0323 PeterH0323 changed the title update-browse_dataset [Improvement] browse_dataset.py Nov 20, 2022
@PeterH0323 PeterH0323 mentioned this pull request Nov 21, 2022
32 tasks
Comment thread tools/analysis_tools/browse_dataset.py
Comment thread tools/analysis_tools/browse_dataset.py
Comment thread tools/analysis_tools/browse_dataset.py Outdated
Comment thread tools/analysis_tools/browse_dataset.py
Comment thread tools/analysis_tools/browse_dataset.py Outdated
Comment thread tools/analysis_tools/browse_dataset.py Outdated
Comment thread tools/analysis_tools/browse_dataset.py Outdated
Comment thread tools/analysis_tools/browse_dataset.py Outdated
Comment thread tools/analysis_tools/browse_dataset.py Outdated
Comment thread tools/analysis_tools/browse_dataset.py
matrixgame2018 and others added 4 commits November 30, 2022 21:03
修改了215行result_i = [result['dataset_sample'] for result in intermediate_imgs]通过lint
@hhaAndroid hhaAndroid merged commit 1f06f4f into open-mmlab:dev Dec 1, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants