moveNet 设置分辨率的代码
时间: 2023-03-31 14:01:52 浏览: 54
以下是 moveNet 设置分辨率的代码:
```python
import cv2
import mediapipe as mp
mp_drawing = mp.solutions.drawing_utils
mp_pose = mp.solutions.pose
# 设置摄像头分辨率
cap = cv2.VideoCapture()
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)
# 初始化 moveNet 模型
with mp_pose.Pose(min_detection_confidence=.5, min_tracking_confidence=.5) as pose:
while cap.isOpened():
success, image = cap.read()
if not success:
print("Ignoring empty camera frame.")
continue
# 转换图像颜色空间
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
# 处理图像
image.flags.writeable = False
results = pose.process(image)
# 绘制姿势估计结果
image.flags.writeable = True
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
mp_drawing.draw_landmarks(image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
# 显示图像
cv2.imshow('MediaPipe moveNet', image)
if cv2.waitKey(5) & xFF == 27:
break
cap.release()
cv2.destroyAllWindows()
```
希望对你有帮助!