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)

效果如下

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

相关推荐
猫天意2 小时前
【目标检测】metrice_curve和loss_curve对比图可视化
人工智能·深度学习·目标检测·计算机视觉·cv
山烛2 小时前
OpenCV:图像透视变换
人工智能·opencv·计算机视觉·图像透视变换
这张生成的图像能检测吗4 小时前
(综述)视觉任务的视觉语言模型
人工智能·计算机视觉·语言模型·自然语言处理·视觉语言模型
星期天要睡觉7 小时前
计算机视觉(opencv)实战二十一——基于 SIFT 和 FLANN 的指纹图像匹配与认证
人工智能·opencv·计算机视觉
星期天要睡觉8 小时前
计算机视觉(opencv)实战十七——图像直方图均衡化
人工智能·opencv·计算机视觉
赴33511 小时前
图像拼接案例,抠图案例
人工智能·python·计算机视觉
Monkey的自我迭代11 小时前
SIFT特征匹配实战:KNN算法实现指纹认证
人工智能·opencv·计算机视觉
【高级技工】11 小时前
立体校正(Stereo Rectification)的原理
图像处理·计算机视觉
大霸王龙12 小时前
基于vLLM与YOLO的智能图像分类系统
yolo·分类·数据挖掘
星期天要睡觉12 小时前
计算机视觉(opencv)实战十八——图像透视转换
人工智能·opencv·计算机视觉