昇腾社区首页
中文
注册

COCO类型目标检测数据集分析

目标检测数据分析内容:分为标注框宽、高等统计值与直方图,像素的对比度等的直方图,横纵坐标中心点散点图等。具体如下:

  1. anchor,overlap统计值。
  2. 标注宽,高,宽高比,面积, 横纵坐标中心点的直方图。
  3. 色相,饱和度,亮度三个参数的平均值直方图,对比度直方图。
  4. 横纵坐标中心点(对应散点图)。

可通过执行“dataset_analysis.py”脚本进行数据集分析,具体参数使用方式参见如下。

表1 dataset_analysis的参数名、类型、取值范围、默认值及说明

参数名

类型

取值范围

默认值

说明

coco_root

str

-

./split_dataset_coco_feature/train

coco数据集路径。

json_file

str

-

instances_annotations.json

coco数据集annotations文件。

result_path

str

-

./result

分析结果保存路径。

data_analysis的命令参考如下:
python3 com_package/object_detection/data_analysis/dataset_analysis.py --coco_root='com_package/object_detection/data_analysis/split_dataset_coco_feature/train' --json_file=instances_annotations.json --result_path='./result'

执行命令参考log信息如下:

图1 data_analysis log信息

脚本执行完毕后会在“--result”参数指定的输出目录中生成标注分析结果。

图2 分析结果