AlphaPose - Формат вывода

Содержание

1. Формат вывода
   1. Порядок ключевых точек
   2. Порядок тепловых карт

Формат вывода

1. По умолчанию мы сохраняем результаты для всех изображений в одном JSON-файле, который аналогичен формату результата, используемому COCO.
   1. keypoints содержит местоположения частей тела и уровень уверенности в детектировании, отформатированный как x1,y1,c1,x2,y2,c2,.... c — это уровень уверенности в диапазоне [0,1] для набора данных MPII и диапазоне [0,6] для набора данных COCO.
   2. score — это уровень уверенности для всего человека, рассчитанный с помощью нашего параметрического алгоритма NMS для поз.

[
 // для человека_1 на изображении_1
 {
    "image_id" : string, image_1_name, 
    "category_id" : int, 1 для человека
    "keypoints" : [x1,y1,c1,...,xk,yk,ck], 
    "score" : float,
 },
 // для человека_2 на изображении_1
 {
    "image_id" : string, image_1_name, 
    "category_id" : int, 1 для человека
    "keypoints" : [x1,y1,c1,...,xk,yk,ck], 
    "score" : float,
 },
 ...
 // для людей на изображении_2
{
    "image_id" : string, image_2_name, 
    "category_id" : int, 1 для человека
    "keypoints" : [x1,y1,c1,...,xk,yk,ck], 
    "score" : float,
 },
 ...
]

2. Если флаг --format установлен как 'cmu', мы будем сохранять результаты для каждого изображения в формате, используемом CMU-Pose.

{
    "version":0.1,
    "bodies":[
        {"joints":[x1,y1,c1,...,xk,yk,ck]},
        {"joints":[x1,y1,c1,...,xk,yk,ck]},
    ]
}
```3. Если флаг `--format` установлен как 'open', мы будем сохранять результаты для каждого изображения в формате, используемом OpenPose.

{ "version": 0.1, "people": [ {"pose_keypoints_2d": [x1, y1, c1, ..., xk, yk, ck]}, {"pose_keypoints_2d": [x1, y1, c1, ..., xk, yk, ck]} ] }

### Порядок ключевых точек
По умолчанию порядок ключевых точек следующий:

// Результат для COCO (17 частей тела) {0, "Нос"}, {1, "Левое глазное отверстие"}, {2, "Правое глазное отверстие"}, {3, "Левое ухо"}, {4, "Правое ухо"}, {5, "Левое плечо"}, {6, "Правое плечо"}, {7, "Левый локоть"}, {8, "Правый локоть"}, {9, "Левое запястье"}, {10, "Правое запястье"}, {11, "Левое бедро"}, {12, "Правое бедро"}, {13, "Левое колено"}, {14, "Правое колено"}, {15, "Левое голень"}, {16, "Правое голень"}, // Результат для MPII (16 частей тела) {0, "Правое голень"}, {1, "Правое колено"}, {2, "Правое бедро"}, {3, "Левое бедро"}, {4, "Левое колено"}, {5, "Левое голень"}, {6, "Подвздошная область"}, {7, "Грудная клетка"}, {8, "Шея"}, {9, "Голова"}, {10, "Правое запястье"}, {11, "Правый локоть"}, {12, "Правое плечо"}, {13, "Левое плечо"}, {14, "Левый локоть"}, {15, "Левое запястье"}

Если флаг `--format` установлен в значение 'cmu' или 'open', то порядок ключевых точек будет следующим:

// Результат для COCO (18 частей тела) {0, "Нос"}, {1, "Шея"}, {2, "Правое плечо"}, {3, "Правый локоть"}, {4, "Правое запястье"}, {5, "Левое плечо"}, {6, "Левый локоть"}, {7, "Левое запястье"}

    {9, "Правое колено"},
    {10, "Правая голень"},
    {11, "Левое бедро"},
    {12, "Левое колено"},
    {13, "Левая голень"},
    {14, "Правое глазное отверстие"},
    {15, "Левое глазное отверстие"},
    {16, "Правое ухо"},
    {17, "Левое ухо"},
    // Результат для MPII (15 частей тела)
    {0, "Голова"},
    {1, "Шея"},
    {2, "Правое плечо"},
    {3, "Правый локоть"},
    {4, "Правое запястье"},
    {5, "Левое плечо"},
    {6, "Левый локоть"},
    {7, "Левое запястье"},
    {8, "Правое бедро"},
    {9, "Правое колено"},
    {10, "Правая голень"},
    {11, "Левое бедро"},
    {12, "Левое колено"},
    {13, "Левая голень"},
    {14, "Грудная клетка"},