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)

效果如下

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

相关推荐
shao9185163 小时前
Gradio全解11——Streaming:流式传输的视频应用(3)——YOLO系列模型技术架构与实战
yolo·coco·yolov10·yoloe
nenchoumi31193 小时前
全网首发!Realsense 全新 D555 相机开箱记录与 D435i、L515、D456 横向测评!
数码相机·计算机视觉·机器人·ros·realsense
小关会打代码3 小时前
计算机视觉之多模板匹配
人工智能·计算机视觉
AI 嗯啦3 小时前
计算机视觉----opencv----身份证号码识别案例
人工智能·opencv·计算机视觉
星期天要睡觉4 小时前
计算机视觉(opencv)——基于模板匹配的信用卡号识别系统
opencv·计算机视觉
荼蘼5 小时前
OpenCV 高阶 图像金字塔 用法解析及案例实现
人工智能·opencv·计算机视觉
CVer儿6 小时前
【天文】星光超分辨图像增强
计算机视觉
星期天要睡觉7 小时前
计算机视觉(opencv)——基于模板匹配的身份证号识别系统
人工智能·opencv·计算机视觉
Francek Chen7 小时前
【深度学习计算机视觉】03:目标检测和边界框
人工智能·pytorch·深度学习·目标检测·计算机视觉·边界框
CoovallyAIHub8 小时前
基于YOLO集成模型的无人机多光谱风电部件缺陷检测
深度学习·算法·计算机视觉