,可以使用列表推导式简化代码:

pose_data = [landmark.x for landmark in results.pose_landmarks.landmark if landmark.visibility >= 0.5 and landmark.name in ['NOSE', 'LEFT_SHOULDER', 'RIGHT_SHOULDER', 'LEFT_ELBOW', 'RIGHT_ELBOW', 'LEFT_WRIST', 'RIGHT_WRIST', 'LEFT_HIP', 'RIGHT_HIP', 'LEFT_KNEE', 'RIGHT_KNEE', 'LEFT_ANKLE', 'RIGHT_ANKLE']] pose_data += [landmark.y for landmark in results.pose_landmarks.landmark if landmark.visibility >= 0.5 and landmark.name in ['NOSE', 'LEFT_SHOULDER', 'RIGHT_SHOULDER', 'LEFT_ELBOW', 'RIGHT_ELBOW', 'LEFT_WRIST', 'RIGHT_WRIST', 'LEFT_HIP', 'RIGHT_HIP', 'LEFT_KNEE', 'RIGHT_KNEE', 'LEFT_ANKLE', 'RIGHT_ANKLE']] pose_data += [landmark.z if landmark.visibility >= 0.5 and landmark.name in ['NOSE', 'LEFT_SHOULDER', 'RIGHT_SHOULDER', 'LEFT_ELBOW', 'RIGHT_ELBOW', 'LEFT_WRIST', 'RIGHT_WRIST', 'LEFT_HIP', 'RIGHT_HIP', 'LEFT_KNEE', 'RIGHT_KNEE', 'LEFT_ANKLE', 'RIGHT_ANKLE'] and landmark.z is not None else 0 for landmark in results.pose_landmarks.landmark

pose_data = for landmark in resultspose_landmarkslandmark if landmarkvisibility 05 # 如果关键点可见度小于05则跳过 continue if landmarkname not in NOSE LEFT_SHOULDER RIGHT_SHOULDER LEFT_ELBOW RIGHT_EL

原文地址: https://www.cveoy.top/t/topic/eyRU 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录