目标检测框架yolov5环境搭建

news/2024/7/11 1:14:38 标签: 目标检测, YOLO, 深度学习, pytorch, yolov5

    目前,目标检测框架中,yolov5 是很火的,它基于pytorch框架,集成opencv等框架,项目地址:https://github.com/ultralytics/yolov5,对我来说,机器学习、深度学习才开始接触,本篇只能是按照项目说明来进行一个简单的测试和验证。

    这里目标检测,其实就是给定一些图片后者视频,结合现有的模型训练,然后生成识别的结果。

    项目介绍给出的两个示例图片如下:

    第一张像是西班牙足球运动员皮克,夏奇拉的前夫弟弟:

    

    还有一个是齐祖打领带咆哮的照片:

 

    这里目标检测,需要识别图片中的人和车,还有领带。

    环境要求:python3.9及以上版本。

    我这里是在linux下,python版本是3.10 。

    1 、克隆github上的项目到本地:https://github.com/ultralytics/yolov5,如果这个地址比较慢,可以考虑国内的码云地址:https://gitee.com/monkeycc/yolov5,这个地址就是同步的github仓库的项目,但是下载速度很快:

git clone https://github.com/ultralytics/yolov5

    2、安装依赖 

cd yolov5
pip install -r requirements.txt

    3、下载模型文件:https://github.com/ultralytics/yolov5/releases/download/v7.0/yolov5s.pt

如果这个地址太慢,可以到这个百度网盘的地址上下载:https://pan.baidu.com/s/1vSqeX7WDyo23FCLOwiTBjA?pwd=yolo 

    下载之后,放到项目根路径下,我们暂时只下载一个模型文件yolov5s.pt。

   4、运行测试demo。

buejee@SKY-20230223WWS:~/yolov5$ python3 detect.py
detect: weights=yolov5s.pt, source=data/images, data=data/coco128.yaml, imgsz=[6                                         40, 640], conf_thres=0.25, iou_thres=0.45, max_det=1000, device=, view_img=False                                         , save_txt=False, save_conf=False, save_crop=False, nosave=False, classes=None,                                          agnostic_nms=False, augment=False, visualize=False, update=False, project=runs/d                                         etect, name=exp, exist_ok=False, line_thickness=3, hide_labels=False, hide_conf=                                         False, half=False, dnn=False, vid_stride=1
YOLOv5 🚀 2023-4-10 Python-3.10.6 torch-2.0.0+cu117 CPU

Fusing layers...
YOLOv5s summary: 213 layers, 7225885 parameters, 0 gradients
image 1/2 /home/buejee/yolov5/data/images/bus.jpg: 640x480 4 persons, 1 bus, 197                                         .4ms
image 2/2 /home/buejee/yolov5/data/images/zidane.jpg: 384x640 2 persons, 2 ties,                                          139.0ms
Speed: 4.6ms pre-process, 168.2ms inference, 9.4ms NMS per image at shape (1, 3,                                          640, 640)
Results saved to runs/detect/exp2
buejee@SKY-20230223WWS:~/yolov5$

     根据提示,结果放到了runs/detect/exp2目录下。

    先看看大致的结果:

    图片中的人都识别出来了,并且用红框画出来了。

    具体的图片:

    第一张,公交车与旁边的人:

 

    第二张,人和领带都识别了:

 

   感觉很强大,很厉害,环境部署也不是很复杂,基本就是傻瓜式的执行几个命令。 


http://www.niftyadmin.cn/n/234190.html

相关文章

RFID技术在智慧图书馆盘点系统中的优势

RFID射频识别及技术,作为一种新兴的非接触式的自动识别技术,其基本原理是电磁理论,因其操作便捷高效,无需人工干预,可在各种恶劣环境下,通过射频信号自动识别目标并获取相关数据,可识别高速运动…

图染色问题的NP完全性证明

文章目录1.Overview2.CNF 3-sat3. Gadgets3.1 Concolorous Edges3.2 Starter/Variable Gadget3.3 Splitter Gadget3.4 OR Gadget3.5 Clause Gadget4. To Planar Graph最近在学 6.890,然后 devans 刚好问了我这个问题,然后尝试编了一个证明。 1.Overview…

【Java EE】-网络编程(一) 网络初识

作者:学Java的冬瓜 博客主页:☀冬瓜的主页🌙 专栏:【JavaEE】 主要内容:单机、局域网、广域网、交换机、路由器。IP地址,端口号,协议,五元组。 协议分层,OSI七层网络模型…

已解决:ModuleNotFoundError: No module named ‘flask._compat‘

📋 个人简介 💖 作者简介:大家好,我是阿牛,全栈领域新星创作者。😜🎉 支持我:点赞👍+收藏⭐️+留言📝📣 系列专栏:flask框架从入门到实战🍁💬格言:要成为光,因为有怕黑的人!🔥 相信很多人在flask项目中都遇到了这个报错:ModuleNotFoundError: No …

Cobalt_Strike_4.5渗透工具的安装与使用

前言: Cobalt Strike是一款内网渗透测试神器,Cobalt Strike分为客户端和服务器端,该服务器端被称为团队服务器,是Beacon有效负载的控制器,同时,cobalt strike也具有社会工程学功能,团队服务器还…

CSS中flex属性的的使用以及应用场景有哪些

文章目录一. flex属性?(虚假的) --- 这里主要是回顾1.1 flex-grow1.2 flex-shrink1.3 flex-basis二. flex属性 ! (真正的!!!)三. flex一些常见的值, 以及使用场景3.1 flex:initial 使用场景3.2 flex:0 和 flex:node 适用场景3.3 flex:1 和 flex:auto3.4 总结一. flex属性?(虚…

用孕妇的思维和孕妇对话——《用孩子的思维和孩子对话》

看惯了某鱼上经典的卖货话术,“不懂事怀孕了,遂出”。没想到我居然也快要当爸爸了。这几个月准爸爸的生活让我切身的体会到了孕妇的不容易以及陪伴孕妇的辛苦。本文不打算歌颂母亲或者替准爸爸们倒苦水,单纯总结思考一下孕期夫妻如何沟通。加…

软件测试,月薪还没过万的进来看

目录 前言 一、技能提升 二、多语言掌握 三、团队协作 四、持续学习 五、提高测试效率 六、质量保障 七、职业发展 八、总结 前言 在当今IT行业中,软件测试是一个不可或缺的环节。随着互联网和移动应用的发展,对软件质量的要求也越来越高。因此…