diff --git a/lite/examples/pose_estimation/raspberry_pi/data.py b/lite/examples/pose_estimation/raspberry_pi/data.py index b72f092cc61..c2507f0b054 100644 --- a/lite/examples/pose_estimation/raspberry_pi/data.py +++ b/lite/examples/pose_estimation/raspberry_pi/data.py @@ -108,7 +108,7 @@ def person_from_keypoints_with_scores( # Calculate person score by averaging keypoint scores. scores_above_threshold = list( filter(lambda x: x > keypoint_score_threshold, scores)) - person_score = np.average(scores_above_threshold) + person_score = np.average(scores_above_threshold) if scores_above_threshold != [] else 0 return Person(keypoints, bounding_box, person_score)