Small filter improvements for mvor dataset.

This commit is contained in:
Daniel
2024-07-03 14:00:09 +02:00
parent 55f7069372
commit 7330b010b5
2 changed files with 110 additions and 110 deletions

View File

@ -820,181 +820,181 @@ Results of the model in various experiments on different datasets.
(duration 00:02:31) (duration 00:02:31)
```json ```json
{ {
"avg_time_2d": 0.05586366622628996, "avg_time_2d": 0.05581602218270494,
"avg_time_3d": 0.0030540410844497805, "avg_time_3d": 0.003066245881729249,
"avg_fps": 16.972826093280727 "avg_fps": 16.983041463445524
} }
{ {
"person_nums": { "person_nums": {
"total_frames": 629, "total_frames": 629,
"total_labels": 1061, "total_labels": 1061,
"total_preds": 1107, "total_preds": 1107,
"considered_empty": 106, "considered_empty": 105,
"valid_preds": 754, "valid_preds": 755,
"invalid_preds": 247, "invalid_preds": 247,
"missing": 307, "missing": 306,
"invalid_fraction": 0.24675, "invalid_fraction": 0.24651,
"precision": 0.75325, "precision": 0.75349,
"recall": 0.71065, "recall": 0.71159,
"f1": 0.73133, "f1": 0.73194,
"non_empty": 1001 "non_empty": 1002
}, },
"mpjpe": { "mpjpe": {
"count": 754, "count": 755,
"mean": 0.112533, "mean": 0.112545,
"median": 0.093582, "median": 0.093641,
"std": 0.064237, "std": 0.064196,
"sem": 0.002341, "sem": 0.002338,
"min": 0.040489, "min": 0.040489,
"max": 0.497874, "max": 0.497874,
"recall-0.025": 0.0, "recall-0.025": 0.0,
"recall-0.05": 0.004713, "recall-0.05": 0.004713,
"recall-0.1": 0.39868, "recall-0.1": 0.39868,
"recall-0.15": 0.602262, "recall-0.15": 0.603205,
"recall-0.25": 0.687088, "recall-0.25": 0.68803,
"recall-0.5": 0.71065, "recall-0.5": 0.711593,
"num_labels": 1061, "num_labels": 1061,
"ap-0.025": 0.0, "ap-0.025": 0.0,
"ap-0.05": 9.5e-05, "ap-0.05": 9.5e-05,
"ap-0.1": 0.257803, "ap-0.1": 0.257626,
"ap-0.15": 0.500043, "ap-0.15": 0.500936,
"ap-0.25": 0.610092, "ap-0.25": 0.611019,
"ap-0.5": 0.638299 "ap-0.5": 0.639232
}, },
"head": { "head": {
"count": 753, "count": 754,
"mean": 0.061147, "mean": 0.061083,
"median": 0.048418, "median": 0.048251,
"std": 0.051506, "std": 0.051502,
"sem": 0.001878, "sem": 0.001877,
"min": 0.006565, "min": 0.006565,
"max": 0.436815, "max": 0.436815,
"recall-0.025": 0.091423, "recall-0.025": 0.092366,
"recall-0.05": 0.36852, "recall-0.05": 0.369463,
"recall-0.1": 0.63525, "recall-0.1": 0.636192,
"recall-0.15": 0.675778, "recall-0.15": 0.67672,
"recall-0.25": 0.69557, "recall-0.25": 0.696513,
"recall-0.5": 0.709708, "recall-0.5": 0.71065,
"num_labels": 1061 "num_labels": 1061
}, },
"shoulder_left": { "shoulder_left": {
"count": 752, "count": 753,
"mean": 0.059702, "mean": 0.059656,
"median": 0.042396, "median": 0.04234,
"std": 0.060756, "std": 0.060728,
"sem": 0.002217, "sem": 0.002215,
"min": 0.002877, "min": 0.002877,
"max": 0.48718, "max": 0.48718,
"recall-0.025": 0.161169, "recall-0.025": 0.162111,
"recall-0.05": 0.424128, "recall-0.05": 0.425071,
"recall-0.1": 0.618285, "recall-0.1": 0.619227,
"recall-0.15": 0.670123, "recall-0.15": 0.671065,
"recall-0.25": 0.689915, "recall-0.25": 0.690858,
"recall-0.5": 0.708765, "recall-0.5": 0.709708,
"num_labels": 1061 "num_labels": 1061
}, },
"shoulder_right": { "shoulder_right": {
"count": 752, "count": 753,
"mean": 0.058725, "mean": 0.058717,
"median": 0.038927, "median": 0.039015,
"std": 0.064616, "std": 0.064574,
"sem": 0.002358, "sem": 0.002355,
"min": 0.001883, "min": 0.001883,
"max": 0.470146, "max": 0.470146,
"recall-0.025": 0.154571, "recall-0.025": 0.154571,
"recall-0.05": 0.454288, "recall-0.05": 0.454288,
"recall-0.1": 0.621112, "recall-0.1": 0.622055,
"recall-0.15": 0.668238, "recall-0.15": 0.66918,
"recall-0.25": 0.685203, "recall-0.25": 0.686145,
"recall-0.5": 0.708765, "recall-0.5": 0.709708,
"num_labels": 1061 "num_labels": 1061
}, },
"elbow_left": { "elbow_left": {
"count": 740, "count": 741,
"mean": 0.100132, "mean": 0.100156,
"median": 0.07137, "median": 0.071628,
"std": 0.082644, "std": 0.08259,
"sem": 0.00304, "sem": 0.003036,
"min": 0.005709, "min": 0.005709,
"max": 0.481769, "max": 0.481769,
"recall-0.025": 0.071631, "recall-0.025": 0.071631,
"recall-0.05": 0.234684, "recall-0.05": 0.234684,
"recall-0.1": 0.428841, "recall-0.1": 0.428841,
"recall-0.15": 0.551367, "recall-0.15": 0.552309,
"recall-0.25": 0.655042, "recall-0.25": 0.655985,
"recall-0.5": 0.697455, "recall-0.5": 0.698398,
"num_labels": 1061 "num_labels": 1061
}, },
"elbow_right": { "elbow_right": {
"count": 744, "count": 745,
"mean": 0.081124, "mean": 0.08108,
"median": 0.056996, "median": 0.056961,
"std": 0.075451, "std": 0.07541,
"sem": 0.002768, "sem": 0.002765,
"min": 0.003759, "min": 0.003759,
"max": 0.498068, "max": 0.498068,
"recall-0.025": 0.093308, "recall-0.025": 0.093308,
"recall-0.05": 0.310085, "recall-0.05": 0.311027,
"recall-0.1": 0.525919, "recall-0.1": 0.526861,
"recall-0.15": 0.617342, "recall-0.15": 0.618285,
"recall-0.25": 0.672008, "recall-0.25": 0.67295,
"recall-0.5": 0.701225, "recall-0.5": 0.702168,
"num_labels": 1061 "num_labels": 1061
}, },
"wrist_left": { "wrist_left": {
"count": 746, "count": 747,
"mean": 0.124492, "mean": 0.124449,
"median": 0.087532, "median": 0.087555,
"std": 0.102378, "std": 0.102316,
"sem": 0.003751, "sem": 0.003746,
"min": 0.004181, "min": 0.004181,
"max": 0.497384, "max": 0.497384,
"recall-0.025": 0.070688, "recall-0.025": 0.070688,
"recall-0.05": 0.190386, "recall-0.05": 0.190386,
"recall-0.1": 0.37606, "recall-0.1": 0.377003,
"recall-0.15": 0.479736, "recall-0.15": 0.480679,
"recall-0.25": 0.609802, "recall-0.25": 0.610745,
"recall-0.5": 0.70311, "recall-0.5": 0.704053,
"num_labels": 1061 "num_labels": 1061
}, },
"wrist_right": { "wrist_right": {
"count": 739, "count": 740,
"mean": 0.110218, "mean": 0.110344,
"median": 0.076683, "median": 0.076747,
"std": 0.097189, "std": 0.097184,
"sem": 0.003578, "sem": 0.003575,
"min": 0.006423, "min": 0.006423,
"max": 0.485199, "max": 0.485199,
"recall-0.025": 0.073516, "recall-0.025": 0.073516,
"recall-0.05": 0.235627, "recall-0.05": 0.235627,
"recall-0.1": 0.430726, "recall-0.1": 0.430726,
"recall-0.15": 0.518379, "recall-0.15": 0.518379,
"recall-0.25": 0.621112, "recall-0.25": 0.622055,
"recall-0.5": 0.696513, "recall-0.5": 0.697455,
"num_labels": 1061 "num_labels": 1061
}, },
"hip_left": { "hip_left": {
"count": 739, "count": 740,
"mean": 0.186663, "mean": 0.186623,
"median": 0.16686, "median": 0.166804,
"std": 0.08462, "std": 0.08457,
"sem": 0.003115, "sem": 0.003111,
"min": 0.021758, "min": 0.021758,
"max": 0.499358, "max": 0.499358,
"recall-0.025": 0.000943, "recall-0.025": 0.000943,
"recall-0.05": 0.004713, "recall-0.05": 0.004713,
"recall-0.1": 0.069746, "recall-0.1": 0.069746,
"recall-0.15": 0.263902, "recall-0.15": 0.263902,
"recall-0.25": 0.574929, "recall-0.25": 0.575872,
"recall-0.5": 0.696513, "recall-0.5": 0.697455,
"num_labels": 1061 "num_labels": 1061
}, },
"hip_right": { "hip_right": {
"count": 742, "count": 743,
"mean": 0.180059, "mean": 0.180336,
"median": 0.164071, "median": 0.164173,
"std": 0.079758, "std": 0.080059,
"sem": 0.00293, "sem": 0.002939,
"min": 0.024563, "min": 0.024563,
"max": 0.492847, "max": 0.492847,
"recall-0.025": 0.000943, "recall-0.025": 0.000943,
@ -1002,7 +1002,7 @@ Results of the model in various experiments on different datasets.
"recall-0.1": 0.074458, "recall-0.1": 0.074458,
"recall-0.15": 0.28558, "recall-0.15": 0.28558,
"recall-0.25": 0.590952, "recall-0.25": 0.590952,
"recall-0.5": 0.69934, "recall-0.5": 0.700283,
"num_labels": 1061 "num_labels": 1061
}, },
"knee_left": {}, "knee_left": {},
@ -1011,18 +1011,18 @@ Results of the model in various experiments on different datasets.
"ankle_right": {}, "ankle_right": {},
"joint_recalls": { "joint_recalls": {
"num_labels": 9549, "num_labels": 9549,
"recall-0.025": 0.07938, "recall-0.025": 0.07959,
"recall-0.05": 0.24673, "recall-0.05": 0.24715,
"recall-0.1": 0.41973, "recall-0.1": 0.42015,
"recall-0.15": 0.5254, "recall-0.15": 0.52581,
"recall-0.25": 0.64321, "recall-0.25": 0.64447,
"recall-0.5": 0.70175 "recall-0.5": 0.70301
} }
} }
{ {
"total_parts": 10610, "total_parts": 10610,
"correct_parts": 5861, "correct_parts": 5868,
"pcp": 0.552403 "pcp": 0.553063
} }
``` ```

View File

@ -306,7 +306,7 @@ def filter_poses(poses3D, poses2D, roomparams, joint_names, drop_few_limbs=True)
if average_length < 0.1: if average_length < 0.1:
drop.append(i) drop.append(i)
continue continue
if average_length > 0.5: if drop_few_limbs and average_length > 0.5:
drop.append(i) drop.append(i)
continue continue