YOLOV8 POSE姿态检测对图片绘制矩形和和关节点序号

代码如下

复制代码
import cv2
import torch

from ultralytics import YOLO

# Load a model
# model = YOLO("yolov8n-pose.yaml")  # build a new model from YAML
model = YOLO("yolov8n-pose.pt")  # load a pretrained model (recommended for training)
# model = YOLO("yolov8n-pose.yaml").load("yolov8n-pose.pt")  # build from YAML and transfer weights

# Train the model
import cv2

cv2.namedWindow('win', cv2.WINDOW_NORMAL)
# cv2.resizeWindow('win', 400, 600)
img_path=r"bus.jpg"
img=cv2.imread(img_path)
results = model(img_path)
for ret in results[0]:
    xmin,ymin,xmax,ymax,conf,cls=ret.boxes.data[0]
    cv2.rectangle(img,(int(xmin),int(ymin)),(int(xmax),int(ymax)),color=(255,0,0),thickness=3,lineType=cv2.LINE_AA)
    keypoints=ret.keypoints.data[0]
    for idx,pt in enumerate(keypoints):
        x,y=int(pt[0]),int(pt[1])
        cv2.circle(img,center=(x,y),radius=2,color=(0,0,255),thickness=2)
        cv2.putText(img,str(idx),org=(x-8,y),fontFace=cv2.FONT_ITALIC,fontScale=0.5,color=(0,0,255),thickness=1)
cv2.imshow("win",img)
cv2.waitKey(0)
cv2.destroyAllWindows()
# print(results)

效果如下

每个可以看得见的关节都标注了序号

相关推荐
jndingxin8 小时前
瑞芯微算法环境搭建(2)------编译opencv
人工智能·opencv·计算机视觉
甜辣uu8 小时前
【源码讲解+复现】YOLOv10: Real-Time End-to-End Object Detection
人工智能·yolo·目标检测·nms-free
星期天要睡觉12 小时前
计算机视觉(opencv)——人脸网格关键点检测
python·opencv·计算机视觉
玉树临风江流儿13 小时前
关于pkg-config的使用示例--g++编译过程引入第三方库(如Opencv、Qt)
人工智能·opencv
Gloria_niki13 小时前
目标检测学习总结
人工智能·计算机视觉·目标跟踪
心无旁骛~15 小时前
PIL与OpenCV图像读取的颜色格式陷阱:RGB vs BGR
人工智能·opencv·计算机视觉
WWZZ202516 小时前
快速上手大模型:机器学习1
人工智能·深度学习·机器学习·计算机视觉·机器人·slam
CoovallyAIHub16 小时前
CLIP, DINO等多模型融合DreamSim,让电脑“看懂”图片有多像!模型融合成为热门!
深度学习·算法·计算机视觉
Giser探索家16 小时前
遥感卫星升轨 / 降轨技术解析:对图像光照、对比度的影响及工程化应用
大数据·人工智能·算法·安全·计算机视觉·分类
Mr数据杨16 小时前
【ComfyUI】Animate单人物角色视频替换
人工智能·计算机视觉·音视频