mirror of https://github.com/open-mmlab/mmyolo.git
63 lines
3.0 KiB
Markdown
63 lines
3.0 KiB
Markdown
|
# 可视化 COCO 标签
|
||
|
|
|
||
|
|
脚本 `tools/analysis_tools/browse_coco_json.py` 能够使用可视化显示 COCO 标签在图片的情况。
|
||
|
|
|
||
|
|
```shell
|
||
|
|
python tools/analysis_tools/browse_coco_json.py [--data-root ${DATA_ROOT}] \
|
||
|
|
[--img-dir ${IMG_DIR}] \
|
||
|
|
[--ann-file ${ANN_FILE}] \
|
||
|
|
[--wait-time ${WAIT_TIME}] \
|
||
|
|
[--disp-all] [--category-names CATEGORY_NAMES [CATEGORY_NAMES ...]] \
|
||
|
|
[--shuffle]
|
||
|
|
```
|
||
|
|
|
||
|
|
其中，如果图片、标签都在同一个文件夹下的话，可以指定 `--data-root` 到该文件夹，然后 `--img-dir` 和 `--ann-file` 指定该文件夹的相对路径，代码会自动拼接。
|
||
|
|
如果图片、标签文件不在同一个文件夹下的话，则无需指定 `--data-root` ，直接指定绝对路径的 `--img-dir` 和 `--ann-file` 即可。
|
||
|
|
|
||
|
|
例子：
|
||
|
|
|
||
|
|
1. 查看 `COCO` 全部类别，同时展示 `bbox`、`mask` 等所有类型的标注：
|
||
|
|
|
||
|
|
```shell
|
||
|
|
python tools/analysis_tools/browse_coco_json.py --data-root './data/coco' \
|
||
|
|
--img-dir 'train2017' \
|
||
|
|
--ann-file 'annotations/instances_train2017.json' \
|
||
|
|
--disp-all
|
||
|
|
```
|
||
|
|
|
||
|
|
如果图片、标签不在同一个文件夹下的话，可以使用绝对路径：
|
||
|
|
|
||
|
|
```shell
|
||
|
|
python tools/analysis_tools/browse_coco_json.py --img-dir '/dataset/image/coco/train2017' \
|
||
|
|
--ann-file '/label/instances_train2017.json' \
|
||
|
|
--disp-all
|
||
|
|
```
|
||
|
|
|
||
|
|
2. 查看 `COCO` 全部类别，同时仅展示 `bbox` 类型的标注，并打乱显示：
|
||
|
|
|
||
|
|
```shell
|
||
|
|
python tools/analysis_tools/browse_coco_json.py --data-root './data/coco' \
|
||
|
|
--img-dir 'train2017' \
|
||
|
|
--ann-file 'annotations/instances_train2017.json' \
|
||
|
|
--shuffle
|
||
|
|
```
|
||
|
|
|
||
|
|
3. 只查看 `bicycle` 和 `person` 类别，同时仅展示 `bbox` 类型的标注：
|
||
|
|
|
||
|
|
```shell
|
||
|
|
python tools/analysis_tools/browse_coco_json.py --data-root './data/coco' \
|
||
|
|
--img-dir 'train2017' \
|
||
|
|
--ann-file 'annotations/instances_train2017.json' \
|
||
|
|
--category-names 'bicycle' 'person'
|
||
|
|
```
|
||
|
|
|
||
|
|
4. 查看 `COCO` 全部类别，同时展示 `bbox`、`mask` 等所有类型的标注，并打乱显示：
|
||
|
|
|
||
|
|
```shell
|
||
|
|
python tools/analysis_tools/browse_coco_json.py --data-root './data/coco' \
|
||
|
|
--img-dir 'train2017' \
|
||
|
|
--ann-file 'annotations/instances_train2017.json' \
|
||
|
|
--disp-all \
|
||
|
|
--shuffle
|
||
|
|
```
|