在Yolov5中使用Mediapipe实现对手部特定范围检测

最新推荐文章于 2024-03-10 17:03:08 发布

王小梦1998

最新推荐文章于 2024-03-10 17:03:08 发布

阅读量1.4k

点赞数 3

文章标签： python 深度学习开发语言

本文链接：https://blog.csdn.net/weixin_42198664/article/details/126681951

版权

该博客介绍了如何利用Yolov5和Mediapipe实现对手部特定范围的物体检测。通过在Yolov5的detect.py中结合Mediapipe获取的手部坐标，可以实现实时地仅对手部周围进行物体检测。代码基于Yolov5的6.0版本，并给出了相应的代码片段，展示了如何在每次获取到手部数据后更新Yolo的检测范围。

摘要由CSDN通过智能技术生成

在Yolov5中使用Mediapipe实现对手部特定范围检测

实现效果如下图所示

使用Yolov5-Mediapipe能够根据手部的坐标变化，实时实现Yolo只对于手部周围检测框内物体的检测，该代码的实现使用的是Yolov5-6.0,链接为https://github.com/ultralytics/yolov5/tags 使用的权重为yolov5s.pt

Yolov5

在这里插入图片描述

效果

在这里插入图片描述

代码

detect.py

在这里插入图片描述由于yolo是逐帧运行的，需要detect在每次获取到手部数据以后需要在Loadstream代码块中对于相机捕获到的手部数据更新

// detect.py
 # Dataloader
    if webcam:
        view_img = check_imshow()
        cudnn.benchmark = True  # set True to speed up constant image size inference
        dataset = LoadStreams(source, img_size=imgsz, stride=stride, auto=pt)
        bs = len(dataset)  # batch_size