pg-coco-pose / config.json
nvriese1's picture
Publish PG-COCO-Pose from Pictograph
8318ea2 verified
Raw History Blame Contribute Delete
5.81 kB
{
"_pictograph": {
"architecture": "RF-DETR Keypoint",
"class_mapping": {
"classes": [
"nose",
"left_eye",
"right_eye",
"left_ear",
"right_ear",
"left_shoulder",
"right_shoulder",
"left_elbow",
"right_elbow",
"left_wrist",
"right_wrist",
"left_hip",
"right_hip",
"left_knee",
"right_knee",
"left_ankle",
"right_ankle"
]
},
"class_names": [
"nose",
"left_eye",
"right_eye",
"left_ear",
"right_ear",
"left_shoulder",
"right_shoulder",
"left_elbow",
"right_elbow",
"left_wrist",
"right_wrist",
"left_hip",
"right_hip",
"left_knee",
"right_knee",
"left_ankle",
"right_ankle"
],
"created_at": "2026-11-05T18:07:26.107133+00:00",
"dataset_name": "COCO 2017 Pose",
"export_name": "COCO 2017 Pose-select",
"frameworks": [
"onnx",
"pytorch"
],
"input_shape": [
576,
576
],
"keypoint_schema": {
"class_names": [
"nose"
],
"keypoint_names": {
"nose": [
"nose",
"left_eye",
"right_eye",
"left_ear",
"right_ear",
"left_shoulder",
"right_shoulder",
"left_elbow",
"right_elbow",
"left_wrist",
"right_wrist",
"left_hip",
"right_hip",
"left_knee",
"right_knee",
"left_ankle",
"right_ankle"
]
},
"num_keypoints_per_class": [
17
],
"skeleton": {
"nose": []
}
},
"model_type": "keypoint_detection",
"name": "PG-COCO-Pose",
"num_classes": 17,
"pipeline": "rfdetr_keypoint",
"precision": "fp32",
"schema_version": 1
},
"training": {
"architecture": "RF-DETR Keypoint",
"augmentation": {
"framework": {},
"ops": [
{
"amount": 0.25,
"op": "noise",
"p": 0.5
},
{
"factor": 0.5,
"op": "contrast",
"p": 0.5
}
]
},
"batch_size": 4,
"classes": [
"nose",
"left_eye",
"right_eye",
"left_ear",
"right_ear",
"left_shoulder",
"right_shoulder",
"left_elbow",
"right_elbow",
"left_wrist",
"right_wrist",
"left_hip",
"right_hip",
"left_knee",
"right_knee",
"left_ankle",
"right_ankle"
],
"early_stopping": true,
"early_stopping_patience": 10,
"epochs": 12,
"grad_accum_steps": 1,
"image_height": 576,
"image_width": 576,
"keypoint_templates": {
"person": {
"edges": [
[
15,
13
],
[
13,
11
],
[
16,
14
],
[
14,
12
],
[
11,
12
],
[
5,
11
],
[
6,
12
],
[
5,
6
],
[
5,
7
],
[
6,
8
],
[
7,
9
],
[
8,
10
],
[
1,
2
],
[
0,
1
],
[
0,
2
],
[
1,
3
],
[
2,
4
],
[
3,
5
],
[
4,
6
]
],
"nodes": [
{
"name": "nose",
"x": 0.5,
"y": 0.119
},
{
"name": "left_eye",
"x": 0.52,
"y": 0.1
},
{
"name": "right_eye",
"x": 0.479,
"y": 0.101
},
{
"name": "left_ear",
"x": 0.551,
"y": 0.109
},
{
"name": "right_ear",
"x": 0.448,
"y": 0.111
},
{
"name": "left_shoulder",
"x": 0.594,
"y": 0.218
},
{
"name": "right_shoulder",
"x": 0.406,
"y": 0.221
},
{
"name": "left_elbow",
"x": 0.64,
"y": 0.354
},
{
"name": "right_elbow",
"x": 0.36,
"y": 0.356
},
{
"name": "left_wrist",
"x": 0.62,
"y": 0.441
},
{
"name": "right_wrist",
"x": 0.383,
"y": 0.436
},
{
"name": "left_hip",
"x": 0.561,
"y": 0.499
},
{
"name": "right_hip",
"x": 0.44,
"y": 0.499
},
{
"name": "left_knee",
"x": 0.566,
"y": 0.697
},
{
"name": "right_knee",
"x": 0.431,
"y": 0.697
},
{
"name": "left_ankle",
"x": 0.566,
"y": 0.9
},
{
"name": "right_ankle",
"x": 0.433,
"y": 0.899
}
]
}
},
"learning_rate": 0.0001,
"model_size": "preview",
"precision": "fp32",
"resolution": 576
}
}