yolov7 姿态识别-人体骨架-实时检测

利用官方发布的yolov7-w6-pose.pt权值文件，调用摄像头进行检测，直接附代码：（将该py文件加入官方代码yolov7-main根目录下，直接运行即可）

import matplotlib.pyplot as plt
import torch
import cv2
from torchvision import transforms
import numpy as np
from utils.datasets import letterbox
from utils.general import non_max_suppression_kpt
from utils.plots import output_to_keypoint, plot_skeleton_kpts# 加载模型
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
weigths = torch.load('yolov7-w6-pose.pt')
model = weigths['model']
model = model.half().to(device)
_ = model.eval()# 读取摄像头画面
url=0
cap = cv2.VideoCapture(url)
while (cap.isOpened()):ret, image = cap.read()#image = cv2.imread('xiaolu.jpg')image = letterbox(image, 960, stride=64, auto=True)[0]image_ = image.copy()image = transforms.ToTensor()(image)image = torch.tensor(np.array([image.numpy()]))image = image.to(device)image = image.half()# 姿势识别with torch.no_grad():output, _ = model(image)output = non_max_suppression_kpt(output, 0.25, 0.65, nc=model.yaml['nc'], nkpt=model.yaml['nkpt'], kpt_label=True)output = output_to_keypoint(output)nimg = image[0].permute(1, 2, 0) * 255nimg = nimg.cpu().numpy().astype(np.uint8)nimg = cv2.cvtColor(nimg, cv2.COLOR_RGB2BGR)for idx in range(output.shape[0]):plot_skeleton_kpts(nimg, output[idx, 7:].T, 3)# 打开摄像头cv2.imshow("ning",nimg)if cv2.waitKey(1) & 0xFF == ord('q'):break
cap.release()
cv2.destroyAllWindows()

yolov7 姿态识别-人体骨架-实时检测相关推荐

yolov7 姿态识别-人体骨架-实时检测+实例分割
(15条消息) YOLOv7实例分割_迷途小书童的Note的博客-CSDN博客https://blog.csdn.net/djstavaV/article/details/126357677 (15条 ...
2D Pose人体关键点实时检测(Python/Android /C++ Demo)
2D Pose人体关键点实时检测(Python/Android /C++ Demo) 目录 2D Pose人体关键点实时检测(Python/Android /C++ Demo) 1.人体关键点数据集 ...
CMU开源：价值百万美元的多目标人体关键点实时检测
相关文章:开源|2017 CVPR(Oral Paper):多目标实时体态估测项目开源主要内容介绍: OpenPose是一个利用OpenCV和Caffe并以C++写成的开源库,用来实现多线程的多人 ...
行为识别人体骨架检测+LSTM
转自http://geek.csdn.net/news/detail/138011 在人工智能研究领域,这一技能叫人体行为识别,是智能监控.人机交互.机器人等诸多应用的一项基础技术.以电影提到的老人智 ...
2D人体骨架检测算法综述
文章目录多人关键点检测多人2d关键点检测的算法(自下而上) 1.OpenPose: Realtime Multi-Person 2D Pose Estimation using Part Affi ...
基于人体骨架检测的安全警戒线
文章目录背景方法人体骨架提取空间关系分析背景在校园安防中学生安全是第一位的,在本篇博文中,将简单介绍一个小demo,楼道栏杆安全检测. 方法第一步骨架提取,第二步人体骨架关键部位与栏杆的 ...
姿态估计：人体骨骼关键点检测综述（2016-2020）
点上方蓝字计算机视觉联盟获取更多干货在右上方 ··· 设为星标 ★,与你不见不散仅作学术分享,不代表本公众号立场,侵权联系删除转载于:作者丨七酱@知乎来源丨https://zhuanlan.z ...
人体骨骼关键点检测综述(2)
五.OpenPose库OpenPose人体姿态识别项目是美国卡耐基梅隆大学(CMU)基于卷积神经网络和监督学习并以Caffe为框架开发的开源库.可以实现人体动作.面部表情.手指运动等姿态估计.适用于单 ...
动态多尺度图表达3D人体骨架运动，实现精准预测效果超SOTA
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达现有的基于3D人体骨架的运动预测方法往往不考虑身体不同部位之间的关 ...

yolov7 姿态识别-人体骨架-实时检测

yolov7 姿态识别-人体骨架-实时检测相关推荐

最新文章

热门文章