Moved pose pre/post-processing into onnx graph.

This commit is contained in:
Daniel
2024-12-03 17:59:09 +01:00
parent 2e5c3399ed
commit 97ff32b9ce
4 changed files with 411 additions and 284 deletions

View File

@ -6,9 +6,9 @@ Results of the model in various experiments on different datasets.
```json
{
"avg_time_2d": 0.01254632634631658,
"avg_time_3d": 0.00036295955463991325,
"avg_fps": 77.4636186441503
"avg_time_2d": 0.02244777841083074,
"avg_time_3d": 0.0003828315411583852,
"avg_fps": 43.800844659994496
}
{
"person_nums": {
@ -27,101 +27,101 @@ Results of the model in various experiments on different datasets.
},
"mpjpe": {
"count": 600,
"mean": 0.066275,
"median": 0.058426,
"std": 0.02768,
"sem": 0.001131,
"min": 0.040807,
"max": 0.188876,
"mean": 0.067837,
"median": 0.059973,
"std": 0.027729,
"sem": 0.001133,
"min": 0.044125,
"max": 0.191545,
"recall-0.025": 0.0,
"recall-0.05": 0.083333,
"recall-0.1": 0.938333,
"recall-0.05": 0.035,
"recall-0.1": 0.931667,
"recall-0.15": 0.95,
"recall-0.25": 1.0,
"recall-0.5": 1.0,
"num_labels": 600,
"ap-0.025": 0.0,
"ap-0.05": 0.011533,
"ap-0.1": 0.899113,
"ap-0.15": 0.915362,
"ap-0.05": 0.003097,
"ap-0.1": 0.889734,
"ap-0.15": 0.915055,
"ap-0.25": 1.0,
"ap-0.5": 1.0
},
"nose": {
"count": 600,
"mean": 0.115024,
"median": 0.099737,
"std": 0.041,
"sem": 0.001675,
"min": 0.02644,
"max": 0.261025,
"mean": 0.116272,
"median": 0.09953,
"std": 0.042967,
"sem": 0.001756,
"min": 0.033845,
"max": 0.263303,
"recall-0.025": 0.0,
"recall-0.05": 0.008333,
"recall-0.1": 0.501667,
"recall-0.15": 0.808333,
"recall-0.25": 0.998333,
"recall-0.1": 0.503333,
"recall-0.15": 0.815,
"recall-0.25": 0.993333,
"recall-0.5": 1.0,
"num_labels": 600
},
"shoulder_left": {
"count": 600,
"mean": 0.034317,
"median": 0.026768,
"std": 0.031799,
"sem": 0.001299,
"min": 0.001234,
"max": 0.178357,
"recall-0.025": 0.456667,
"mean": 0.034881,
"median": 0.027327,
"std": 0.031594,
"sem": 0.001291,
"min": 0.002162,
"max": 0.178271,
"recall-0.025": 0.438333,
"recall-0.05": 0.863333,
"recall-0.1": 0.946667,
"recall-0.15": 0.966667,
"recall-0.15": 0.963333,
"recall-0.25": 1.0,
"recall-0.5": 1.0,
"num_labels": 600
},
"shoulder_right": {
"count": 600,
"mean": 0.047981,
"median": 0.034263,
"std": 0.039767,
"sem": 0.001625,
"min": 0.005363,
"max": 0.24597,
"recall-0.025": 0.226667,
"recall-0.05": 0.743333,
"recall-0.1": 0.916667,
"recall-0.15": 0.948333,
"mean": 0.050288,
"median": 0.03555,
"std": 0.042274,
"sem": 0.001727,
"min": 0.003983,
"max": 0.238328,
"recall-0.025": 0.176667,
"recall-0.05": 0.748333,
"recall-0.1": 0.9,
"recall-0.15": 0.94,
"recall-0.25": 1.0,
"recall-0.5": 1.0,
"num_labels": 600
},
"elbow_left": {
"count": 600,
"mean": 0.043526,
"median": 0.034276,
"std": 0.034786,
"sem": 0.001421,
"min": 0.003312,
"max": 0.198715,
"recall-0.025": 0.24,
"recall-0.05": 0.781667,
"recall-0.1": 0.943333,
"recall-0.15": 0.958333,
"mean": 0.044326,
"median": 0.035816,
"std": 0.034654,
"sem": 0.001416,
"min": 0.001741,
"max": 0.198882,
"recall-0.025": 0.226667,
"recall-0.05": 0.776667,
"recall-0.1": 0.946667,
"recall-0.15": 0.955,
"recall-0.25": 1.0,
"recall-0.5": 1.0,
"num_labels": 600
},
"elbow_right": {
"count": 600,
"mean": 0.04376,
"median": 0.033219,
"std": 0.037126,
"sem": 0.001517,
"min": 0.006159,
"max": 0.314756,
"recall-0.025": 0.245,
"recall-0.05": 0.803333,
"recall-0.1": 0.933333,
"mean": 0.044545,
"median": 0.033152,
"std": 0.037755,
"sem": 0.001543,
"min": 0.008169,
"max": 0.338555,
"recall-0.025": 0.218333,
"recall-0.05": 0.798333,
"recall-0.1": 0.928333,
"recall-0.15": 0.943333,
"recall-0.25": 0.996667,
"recall-0.5": 1.0,
@ -129,146 +129,146 @@ Results of the model in various experiments on different datasets.
},
"wrist_left": {
"count": 600,
"mean": 0.044151,
"median": 0.026578,
"std": 0.047109,
"sem": 0.001925,
"min": 0.002328,
"max": 0.288425,
"recall-0.025": 0.478333,
"recall-0.05": 0.736667,
"recall-0.1": 0.883333,
"recall-0.15": 0.918333,
"recall-0.25": 0.998333,
"mean": 0.044896,
"median": 0.025929,
"std": 0.048601,
"sem": 0.001986,
"min": 0.002701,
"max": 0.326901,
"recall-0.025": 0.476667,
"recall-0.05": 0.735,
"recall-0.1": 0.885,
"recall-0.15": 0.913333,
"recall-0.25": 0.996667,
"recall-0.5": 1.0,
"num_labels": 600
},
"wrist_right": {
"count": 600,
"mean": 0.045218,
"median": 0.026994,
"std": 0.050213,
"sem": 0.002052,
"min": 0.002207,
"max": 0.291549,
"recall-0.025": 0.471667,
"recall-0.05": 0.766667,
"recall-0.1": 0.876667,
"recall-0.15": 0.908333,
"recall-0.25": 0.998333,
"mean": 0.045586,
"median": 0.027856,
"std": 0.048323,
"sem": 0.001974,
"min": 0.001841,
"max": 0.229728,
"recall-0.025": 0.436667,
"recall-0.05": 0.751667,
"recall-0.1": 0.881667,
"recall-0.15": 0.916667,
"recall-0.25": 1.0,
"recall-0.5": 1.0,
"num_labels": 600
},
"hip_left": {
"count": 600,
"mean": 0.085085,
"median": 0.079726,
"std": 0.032918,
"sem": 0.001345,
"min": 0.020039,
"max": 0.232252,
"recall-0.025": 0.006667,
"recall-0.05": 0.055,
"recall-0.1": 0.853333,
"recall-0.15": 0.95,
"mean": 0.087757,
"median": 0.083491,
"std": 0.032627,
"sem": 0.001333,
"min": 0.004177,
"max": 0.235198,
"recall-0.025": 0.008333,
"recall-0.05": 0.031667,
"recall-0.1": 0.851667,
"recall-0.15": 0.953333,
"recall-0.25": 1.0,
"recall-0.5": 1.0,
"num_labels": 600
},
"hip_right": {
"count": 600,
"mean": 0.108514,
"median": 0.106487,
"std": 0.025557,
"sem": 0.001044,
"min": 0.043182,
"max": 0.228959,
"mean": 0.112758,
"median": 0.11174,
"std": 0.025369,
"sem": 0.001037,
"min": 0.057593,
"max": 0.231402,
"recall-0.025": 0.0,
"recall-0.05": 0.003333,
"recall-0.1": 0.358333,
"recall-0.15": 0.948333,
"recall-0.05": 0.0,
"recall-0.1": 0.251667,
"recall-0.15": 0.945,
"recall-0.25": 1.0,
"recall-0.5": 1.0,
"num_labels": 600
},
"knee_left": {
"count": 600,
"mean": 0.063226,
"median": 0.046357,
"std": 0.059854,
"sem": 0.002446,
"min": 0.016702,
"max": 0.474089,
"recall-0.025": 0.035,
"recall-0.05": 0.563333,
"recall-0.1": 0.881667,
"count": 599,
"mean": 0.063316,
"median": 0.047979,
"std": 0.058253,
"sem": 0.002382,
"min": 0.019525,
"max": 0.476803,
"recall-0.025": 0.033333,
"recall-0.05": 0.546667,
"recall-0.1": 0.89,
"recall-0.15": 0.925,
"recall-0.25": 0.98,
"recall-0.5": 1.0,
"recall-0.25": 0.978333,
"recall-0.5": 0.998333,
"num_labels": 600
},
"knee_right": {
"count": 600,
"mean": 0.050723,
"median": 0.041264,
"std": 0.036826,
"sem": 0.001505,
"min": 0.013747,
"max": 0.274,
"recall-0.025": 0.043333,
"recall-0.05": 0.745,
"recall-0.1": 0.945,
"mean": 0.050955,
"median": 0.041526,
"std": 0.037031,
"sem": 0.001513,
"min": 0.005291,
"max": 0.27011,
"recall-0.025": 0.035,
"recall-0.05": 0.746667,
"recall-0.1": 0.943333,
"recall-0.15": 0.945,
"recall-0.25": 0.996667,
"recall-0.25": 0.998333,
"recall-0.5": 1.0,
"num_labels": 600
},
"ankle_left": {
"count": 600,
"mean": 0.097084,
"median": 0.085682,
"std": 0.046353,
"sem": 0.001894,
"min": 0.045955,
"max": 0.492226,
"count": 598,
"mean": 0.097897,
"median": 0.086817,
"std": 0.048343,
"sem": 0.001979,
"min": 0.048922,
"max": 0.493127,
"recall-0.025": 0.0,
"recall-0.05": 0.001667,
"recall-0.1": 0.833333,
"recall-0.05": 0.003333,
"recall-0.1": 0.83,
"recall-0.15": 0.933333,
"recall-0.25": 0.985,
"recall-0.5": 1.0,
"recall-0.25": 0.98,
"recall-0.5": 0.996667,
"num_labels": 600
},
"ankle_right": {
"count": 599,
"mean": 0.082224,
"median": 0.068812,
"std": 0.047465,
"sem": 0.001941,
"min": 0.029154,
"max": 0.404964,
"mean": 0.084814,
"median": 0.07029,
"std": 0.053839,
"sem": 0.002202,
"min": 0.025955,
"max": 0.384465,
"recall-0.025": 0.0,
"recall-0.05": 0.026667,
"recall-0.1": 0.888333,
"recall-0.15": 0.91,
"recall-0.25": 0.985,
"recall-0.05": 0.02,
"recall-0.1": 0.886667,
"recall-0.15": 0.908333,
"recall-0.25": 0.973333,
"recall-0.5": 0.998333,
"num_labels": 600
},
"joint_recalls": {
"num_labels": 7800,
"recall-0.025": 0.16923,
"recall-0.05": 0.46833,
"recall-0.1": 0.82692,
"recall-0.15": 0.92692,
"recall-0.25": 0.99487,
"recall-0.5": 0.99974
"recall-0.025": 0.15718,
"recall-0.05": 0.46321,
"recall-0.1": 0.81846,
"recall-0.15": 0.92654,
"recall-0.25": 0.99308,
"recall-0.5": 0.99923
}
}
{
"total_parts": 8400,
"correct_parts": 8097,
"pcp": 0.963929
"correct_parts": 8077,
"pcp": 0.961548
}
```