|
2 | 2 | ## 目录 |
3 | 3 | - [COCO数据集](#COCO数据集) |
4 | 4 | - [MPII数据集](#MPII数据集) |
| 5 | +- [训练其他数据集](#训练其他数据集) |
5 | 6 |
|
6 | 7 | ## COCO数据集 |
7 | 8 | ### COCO数据集的准备 |
@@ -76,32 +77,63 @@ MPII keypoint indexes:: |
76 | 77 | 下面以一个解析后的标注信息为例,说明标注的内容,其中每条标注信息标注了一个人物实例: |
77 | 78 | ``` |
78 | 79 | { |
79 | | - 'joints_vis': [0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1], |
80 | | - 'joints': [ |
81 | | - [-1.0, -1.0], |
82 | | - [-1.0, -1.0], |
83 | | - [-1.0, -1.0], |
84 | | - [-1.0, -1.0], |
85 | | - [-1.0, -1.0], |
86 | | - [-1.0, -1.0], |
87 | | - [-1.0, -1.0], |
88 | | - [1232.0, 288.0], |
89 | | - [1236.1271, 311.7755], |
90 | | - [1181.8729, -0.77553], |
91 | | - [692.0, 464.0], |
92 | | - [902.0, 417.0], |
93 | | - [1059.0, 247.0], |
94 | | - [1405.0, 329.0], |
95 | | - [1498.0, 613.0], |
96 | | - [1303.0, 562.0] |
97 | | - ], |
98 | | - 'image': '077096718.jpg', |
99 | | - 'scale': 9.516749, |
100 | | - 'center': [1257.0, 297.0] |
| 80 | + 'joints_vis': [0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1], |
| 81 | + 'joints': [ |
| 82 | + [-1.0, -1.0], |
| 83 | + [-1.0, -1.0], |
| 84 | + [-1.0, -1.0], |
| 85 | + [-1.0, -1.0], |
| 86 | + [-1.0, -1.0], |
| 87 | + [-1.0, -1.0], |
| 88 | + [-1.0, -1.0], |
| 89 | + [1232.0, 288.0], |
| 90 | + [1236.1271, 311.7755], |
| 91 | + [1181.8729, -0.77553], |
| 92 | + [692.0, 464.0], |
| 93 | + [902.0, 417.0], |
| 94 | + [1059.0, 247.0], |
| 95 | + [1405.0, 329.0], |
| 96 | + [1498.0, 613.0], |
| 97 | + [1303.0, 562.0] |
| 98 | + ], |
| 99 | + 'image': '077096718.jpg', |
| 100 | + 'scale': 9.516749, |
| 101 | + 'center': [1257.0, 297.0] |
101 | 102 | } |
102 | 103 | ``` |
103 | 104 | - `joints_vis`:分别表示16个关键点是否标注,若为0,则对应序号的坐标也为`[-1.0, -1.0]`。 |
104 | 105 | - `joints`:分别表示16个关键点的坐标。 |
105 | 106 | - `image`:表示对应的图片文件。 |
106 | 107 | - `center`:表示人物的大致坐标,用于定位人物在图像中的位置。 |
107 | | -- `scale`:表示人物的比例,对应200px。 |
| 108 | +- `scale`:表示人物的比例,对应200px。 |
| 109 | + |
| 110 | + |
| 111 | +## 训练其他数据集 |
| 112 | +这里我们以`AIChallenger`数据集为例,展示如何将其他数据集对齐到COCO格式并加入关键点模型训练中。 |
| 113 | + |
| 114 | + |
| 115 | +`AI challenger`的标注格式如下: |
| 116 | +``` |
| 117 | +AI Challenger Description: |
| 118 | + 0: 'Right Shoulder', |
| 119 | + 1: 'Right Elbow', |
| 120 | + 2: 'Right Wrist', |
| 121 | + 3: 'Left Shoulder', |
| 122 | + 4: 'Left Elbow', |
| 123 | + 5: 'Left Wrist', |
| 124 | + 6: 'Right Hip', |
| 125 | + 7: 'Right Knee', |
| 126 | + 8: 'Right Ankle', |
| 127 | + 9: 'Left Hip', |
| 128 | + 10: 'Left Knee', |
| 129 | + 11: 'Left Ankle', |
| 130 | + 12: 'Head top', |
| 131 | + 13: 'Neck' |
| 132 | +``` |
| 133 | +1. 将`AI Challenger`点位序号,调整至与`COCO`数据集一致,(如`Right Shoulder`的序号由`0`调整到`13`。 |
| 134 | +2. 统一是否标注/可见的标志位信息,如`AI Challenger`中`标注且可见`需要由`1`调整到`2`。 |
| 135 | +3. 在该过程中,舍弃该数据集特有的点位(如`Neck`);同时该数据集中没有的COCO点位(如`left_eye`等),对应设置为`v=0, x=0, y=0`,表示该未标注。 |
| 136 | +4. 为了避免不同数据集ID重复的问题,需要重新排列图像的`image_id`和`annotation id`。 |
| 137 | +5. 整理图像路径`file_name`,使其能够被正确访问到。 |
| 138 | + |
| 139 | +我们提供了整合`COCO`训练集和`AI Challenger`数据集的[标注文件](https://bj.bcebos.com/v1/paddledet/data/keypoint/aic_coco_train_cocoformat.json),供您参考调整后的效果。 |
0 commit comments