YOLOv8 全家桶再迎新成员!新增Pose Estimation模型!

这篇具有很好参考价值的文章主要介绍了YOLOv8 全家桶再迎新成员!新增Pose Estimation模型!。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

关注公众号,发现CV技术之美

不知不觉间,YOLOv8已经发布三个月了,等待中的YOLOv8论文没来,昨天官方默默又加了新模型:姿态估计。

说好的"目标检测"工业界标杆,正向着“CV全家桶”阔步向前。

现在你可以用YOLOv8做目标检测、实例分割、图像分类、目标跟踪、姿态估计了,也许还有更多惊喜在后面。

要想使用最新的姿态估计功能,你需要更新到最新版的YOLOv8:

pip install --upgrade ultralytics

官方的模型可以在这里下载:

https://github.com/ultralytics/assets/releases

其实你也可以不用下载,如果你仅调用官方模型,程序运行时没找到模型的话,它会自己下载的,就是这么贴心!(不过要保证你的网络良好,亲测国内略慢)

模型是在COCO数据集训练的,目前支持的是人体的检测和姿态估计。

CV君用一张网络图片测试一下:

yolo pose predict model=yolov8n-pose.pt source='http://neweuropeans.org/wp-content/uploads/2016/01/2015_Chisinau_Crossing_Europe_Poike-Stomps_MG_1047.jpg' show=True save=True

其中pose指定任务类型,predict代表我们是要做推断,模型这里我选择的是最轻量级的YOLOv8n-pose,”show=True save=True“代表显示并保存。

运行结果:

Ultralytics YOLOv8.0.68  Python-3.9.13 torch-1.13.1+cu117 CUDA:0 (NVIDIA GeForce GTX 1080 Ti, 11264MiB)
YOLOv8n-pose summary (fused): 187 layers, 3289964 parameters, 0 gradients, 9.2 GFLOPs

Downloading http:\neweuropeans.org\wp-content\uploads\2016\01\2015_Chisinau_Crossing_Europe_Poike-Stomps_MG_1047.jpg to 2015_Chisinau_Crossing_Europe_Poike-Stomps_MG_1047.jpg...
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 564k/564k [00:00<00:00, 1.29MB/s]
image 1/1 E:\download\2015_Chisinau_Crossing_Europe_Poike-Stomps_MG_1047.jpg: 448x640 17 persons, 24.6ms
Speed: 0.0ms preprocess, 24.6ms inference, 7.0ms postprocess per image at shape (1, 3, 640, 640)
Results saved to runs\pose\predict
YOLOv8 全家桶再迎新成员!新增Pose Estimation模型!

我的GPU是1080Ti,这张1500x977大小的图片推断时间24.6ms。

如果要用YOLOv8调用摄像头的话,也非常简单:

yolo pose predict model=yolov8n-pose.pt source=0 show=True save=True

官方在COCO数据集上做了更多测试:

YOLOv8 全家桶再迎新成员!新增Pose Estimation模型!

总计发布了YOLOv8n-pose、YOLOv8s-pose...YOLOv8x-pose-p6等6个模型,在A100上的推断速度从1.18ms到10.04ms,模型参数3.3M到99.1M。

方便在不同硬件和算力的平台上使用。

当然从CPU上的测试结果看,还不是一个CPU实时的算法。

不过更有价值的可能是,要训练我们自己的模型,是非常方便的。

按照coco128-pose.yaml的"样例"组织好数据并修改coco128-pose.yaml后,你只需要一句命令:

yolo pose train data=coco128-pose.yaml model=yolov8n-pose.pt epochs=100 imgsz=640

相关文档:https://docs.ultralytics.com/tasks/pose/

写在最后:YOLOv8的功能越来越多,而且相比于其他开源库,对于工业界来说更友好,涵盖训练、评估、推断、部署全流程,是快速进行项目开发的首选。

下一个进入YOLOv8 "CV全家桶"会是谁呢?Face?

END,入群👇备注:目标检测文章来源地址https://www.toymoban.com/news/detail-439045.html

到了这里,关于YOLOv8 全家桶再迎新成员!新增Pose Estimation模型!的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 以yolov8-pose为案例学习如何写deepstream的回调函数

    在给定的代码中,使用了以下几个元素来构建 DeepStream 管道: source :这个元素是用于从文件中读取视频流的输入源。它可以是任何 GStreamer 支持的视频输入源,如文件、摄像头、网络流等。 streammux :这个元素是流复用器,用于将多个流合并成一个流,并将多帧画面打包为批

    2024年02月13日
    浏览(30)
  • SynergyNet(头部姿态估计 Head Pose Estimation)复现 demo测试

    SynergyNet(github):https://github.com/choyingw/SynergyNet b站:https://www.bilibili.com/video/BV1Ph4y1k7LK/ 我用的AutoDL平台搭建 选择镜像 PyTorch 1.9.0 Python 3.8(ubuntu18.04) Cuda 11.1 安装其它包: 备注:如果无法下载,就到github中下载压缩包,然后上传解压安装 或者使用码云下载: 需要下载如下文件

    2024年02月14日
    浏览(34)
  • 主干网络篇 | YOLOv8 更换主干网络之 VanillaNet |《华为方舟实验室最新成果》| 新增多个尺寸

    2023/11/14 新增 VanillaNet5 VanillaNet13 🍀 论文地址:https://arxiv.org/pdf/2305.12972.pdf 代码地址:https://github.com/huawei-noah/VanillaNet 在基础模型的核心是“多样性即不同”,这一哲学在计算机视觉和自然语言处理方面取得了惊人的成功。然而,优化和Transformer模型固有的复杂性带来了挑战

    2024年02月04日
    浏览(45)
  • 计算机视觉算法中的 相机姿态估计(Camera Pose Estimation)

    目录 ​编辑引言 相机姿态估计的基本概念 相机姿态估计的方法 特征点匹配 直接法 基于深度学习的方法 相机姿态估计的应用 增强现实(AR) 机器人导航 三维重建 结论 相机姿态估计是计算机视觉领域的重要任务之一。它涉及到确定相机在三维空间中的位置和朝向,常用于

    2024年02月05日
    浏览(40)
  • Human Pose Regression with Residual Log-likelihood Estimation

            通过似然热图对输出分布进行建模的基于热图的方法在人体姿态估计领域占据主导地位。相比之下,基于回归的方法更有效,但效果较差。 在这项工作中,我们探索了最大似然估计(MLE),以开发一种高效有效的基于回归的方法。从MLE的角度来看,采用不同的回

    2023年04月26日
    浏览(39)
  • End-to-end 3D Human Pose Estimation with Transformer

    基于Transformer的端到端三维人体姿态估计 基于Transformer的架构已经成为自然语言处理中的常见选择,并且现在正在计算机视觉任务中实现SOTA性能,例如图像分类,对象检测。然而,卷积方法在3D人体姿态估计的许多方法中仍然保持SOTA性能。受视觉变换器最近发展的启发,我们

    2024年02月08日
    浏览(52)
  • 实验:如何在YOLO8中添加PnP算法来实现Head Pose Estimation

    YOLO8的集成度比较高,如何在简洁的代码中加入Head Pose的东西,不是一件简单的事情.这里介绍如何插入PnP算法实现头部姿态估计的代码? 修改predict中的模型,改为我们自己训练的YOLOLandmark模型. 修改cfg入口,由于我们修改来config中不少超参数,因此我们干脆修改来为fi

    2024年02月16日
    浏览(32)
  • Self-supervised 3D Human Pose Estimation from a Single Image

    基于单幅图像的自监督三维人体姿态估计   主页: https://josesosajs.github.io/ imagepose/ 源码:未开源 我们提出了一种新的自我监督的方法预测三维人体姿势从一个单一的图像。预测网络是从描绘处于典型姿势的人的未标记图像的数据集和一组未配对的2D姿势训练的。通过最小化

    2024年02月10日
    浏览(36)
  • 基于人体姿势估计的舞蹈检测(AI Dance based on Human Pose Estimation)

    人体姿势骨架以图形格式表示人的方向。本质上,它是一组坐标,可以连接起来描述人的姿势。骨架中的每个坐标都被称为一个部分(或一个关节,或一个关键点)。两个部分之间的有效连接称为一对(或分支)。下面是一个人体姿势骨架样本。 因此,在本文中,我们将研究如何

    2024年02月11日
    浏览(31)
  • 手撕 视觉slam14讲 ch7 / pose_estimation_3d2d.cpp (1)

     1. 读图,两张rgb(cv::imread)  2. 找到两张rgb图中的特征点匹配对        2.1定义所需要的参数:keypoints1, keypoints2,matches        2.2 提取每张图像的检测 Oriented FAST 角点位置并匹配筛选(调用功能函数1)  3. 建立3d点(像素坐标到相机坐标)         3.1读出深度图(c

    2024年02月10日
    浏览(33)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包