fix criterion crash for poor visibility

2022-09-22 16:36:02 +00:00 · 2022-09-22 16:36:02 +00:00 · e07976945f
commit e07976945f
parent 3c6ddcf7a3
1 changed files with 2 additions and 0 deletions
--- a/easymocap/assignment/criterion.py
+++ b/easymocap/assignment/criterion.py
@ -76,6 +76,8 @@ class CritMinMax(BaseCrit):
    def __call__(self, keypoints3d, **kwargs):
        idx = keypoints3d[..., -1] > self.min_conf
        k3d = keypoints3d[idx, :3]
+        if sum(idx) == 0:
+            return False
        mink = np.min(k3d, axis=0)
        maxk = np.max(k3d, axis=0)
        length = max(np.abs(maxk - mink))